Pin python OT version (#146)

atong01 · Nov 17, 2024 · 62c44af · 62c44af
1 parent 72ae2fd
commit 62c44af
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
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",