From 37d68a1c2ad5a6d181063581daf1749570c684e1 Mon Sep 17 00:00:00 2001
From: Yoshiaki Bando <yoshipon@users.noreply.github.com>
Date: Thu, 11 Apr 2024 21:11:23 +0900
Subject: [PATCH] add dependencies

---
 pyproject.toml | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 12f4c68f2..92ecdc2a0 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -16,10 +16,11 @@ classifiers = [
     "License :: OSI Approved :: MIT License",
 ]
 dependencies = [
+    "colorama",
     "numpy",
-    "lightning",
+    "lightning>=2.2.1",
     "hydra-core",
-    "torch",
+    "torch>=2.2.0",
     "omegaconf",
     "h5py"
 ]