Update pyproject.toml

automl · Jun 18, 2024 · 4f41366 · 4f41366
1 parent 60adad3
commit 4f41366
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -56,7 +56,7 @@ 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"
+torch = ">1.7.0,!=2.0.1, !=2.1.0"
 matplotlib = "^3"
 more-itertools = "*"
 portalocker = "^2"