fixing lightning<=2.2.5 for now

Signed-off-by: CarlosGomes98 <[email protected]>
IBM · Jun 28, 2024 · c2064f6 · c2064f6
1 parent 3481674
commit c2064f6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -39,7 +39,8 @@ dependencies = [
   "lightly>=1.4.25",
   "h5py>=3.10.0",
   "geobench>=1.0.0",
-  "mlflow>=2.12.1"
+  "mlflow>=2.12.1",
+  "lightning<=2.2.5"
 ]
 
 [project.optional-dependencies]