Update pyproject.toml

Toloka · Jan 13, 2025 · a90bdca · a90bdca
1 parent 6916103
commit a90bdca
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -75,7 +75,8 @@ docs = [
     "mkdocstrings-python"
 ]
 learning = [
-    "torch >= 1.6.0",
+    "torch >= 1.6.0,< 2.5.0; python_version < '3.9'",
+    "torch >= 2.5.0; python_version >= '3.9'",
     "transformers >= 4.0.0"
 ]
 
@@ -104,9 +105,3 @@ src_paths = ["crowdkit", "tests"]
 
 [tool.pyupgrade]
 py38-plus = true
-
-[tool.uv]
-override-dependencies = [
-    "torch < 2.5.0; python_version < '3.9'",
-    "torch >= 2.5.0; python_version >= '3.9'"
-]