unpin numpy,pandas and pot versions

atong01 · Jan 3, 2025 · 06471fb · 06471fb
1 parent b0e062b
commit 06471fb
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 10 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -3,13 +3,13 @@ torchvision>=0.11.0
 
 lightning-bolts
 matplotlib
-numpy<2.0.0
+numpy
 scipy
 scikit-learn
 scprep
 scanpy
 torchdyn>=1.0.6   # 1.0.4 is broken on pypi
-pot<0.9.5
+pot
 torchdiffeq
 absl-py
 clean-fid
diff --git a/setup.py b/setup.py
@@ -6,19 +6,14 @@
 
 install_requires = [
     "torch>=1.11.0",
-    "torchvision>=0.11.0",
-    "lightning-bolts",
     "matplotlib",
-    "numpy<2.0.0",  # Due to pandas incompatibility
+    "numpy",  # Due to pandas incompatibility
     "scipy",
     "scikit-learn",
-    "scprep",
-    "scanpy",
-    "torchdyn",
-    "pot<0.9.5",
+    "torchdyn>=1.0.6",
+    "pot",
     "torchdiffeq",
     "absl-py",
-    "clean-fid",
 ]
 
 version_py = os.path.join(os.path.dirname(__file__), "torchcfm", "version.py")