pin pytorch

angelolab · Apr 24, 2024 · 1572f9e · 1572f9e
1 parent 04b8412
commit 1572f9e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -36,7 +36,8 @@ dependencies = [
     "spatial-lda>=0.1.3,<1",
     "statsmodels>=0.13.2,<1",
     "tifffile>=2022",
-    "torchdata<=2.2.2",
+    "torchdata",
+    "pytorch<=2.2.2",
     "tqdm>=4,<5",
     "umap-learn>=0.5,<1.0",
     "xarray>=2022",