Testing version for smp

Signed-off-by: João Lucas de Sousa Almeida <[email protected]>
IBM · Aug 2, 2024 · 80d655c · 80d655c
1 parent 6c483a8
commit 80d655c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -41,7 +41,8 @@ dependencies = [
   # broken due to https://github.com/Lightning-AI/pytorch-lightning/issues/19977
   "lightning>=2, <=2.2.5",
   # see issue #64
-  "albumentations<=1.4.10"
+  "albumentations<=1.4.10",
+  "segmentation-models-pytorch>=0.3.3"
 ]
 
 [project.optional-dependencies]

diff --git a/requirements/required.txt b/requirements/required.txt
@@ -10,4 +10,5 @@ lightly==1.4.25
 h5py==3.10.0
 geobench==1.0.0
 mlflow==2.14.3
-lightning==2.2.5
+lightning==2.2.5
+segmentation-models-pytorch==0.3.3