Update pyproject.toml for PyTorch versions

automl · Jun 18, 2024 · 60adad3 · 60adad3
1 parent 0553bd5
commit 60adad3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -55,7 +55,8 @@ pandas = "^2"
 networkx = "^2.6.3"
 nltk = "^3.6.4"
 scipy = "^1"
-torch = ">=1.7.0,<=2.1, !=2.0.1, !=2.1.0" # fix from: https://stackoverflow.com/a/76647180
+# torch = ">=1.7.0,<=2.1, !=2.0.1, !=2.1.0" # fix from: https://stackoverflow.com/a/76647180
+torch = ">1.7.0"
 matplotlib = "^3"
 more-itertools = "*"
 portalocker = "^2"