bound torch version from above

orbital-materials · Dec 19, 2024 · 0b3571d · 0b3571d
1 parent d7774d3
commit 0b3571d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -20,7 +20,7 @@ dependencies = [
     "ase>=3.23.0",
     "numpy>=1.26.4",
     "scipy>=1.13.1",
-    "torch>=2.2.0",
+    "torch>=2.2.0, <3.0.0",
     "dm-tree>=0.1.8",
     "tqdm>=4.66.5",
 ]