Pin python OT version (#146)

atong01 · web-flow · commit 62c44affd877 · 2024-11-17T13:01:48.000Z
diff --git a/requirements.txt b/requirements.txt
@@ -9,7 +9,7 @@ scikit-learn
 scprep
 scanpy
 torchdyn>=1.0.6   # 1.0.4 is broken on pypi
-pot
+pot<0.9.5
 torchdiffeq
 absl-py
 clean-fid
diff --git a/runner-requirements.txt b/runner-requirements.txt
@@ -45,7 +45,7 @@ scprep
 scanpy
 timm
 torchdyn>=1.0.5   # 1.0.4 is broken on pypi
-pot
+pot<0.9.5
 
 # --------- notebook reqs -------- #
 seaborn>=0.12.2
diff --git a/setup.py b/setup.py
@@ -15,7 +15,7 @@
     "scprep",
     "scanpy",
     "torchdyn",
-    "pot",
+    "pot<0.9.5",
     "torchdiffeq",
     "absl-py",
     "clean-fid",
