Bump pytorch version because <=2.1 does not have cpuonly py312 wheels

flexcompute · Dec 11, 2024 · cce91bb · cce91bb
1 parent 441b504
commit cce91bb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -80,8 +80,8 @@ pyswarms = { version = "*", optional = true }
 
 # pytorch
 torch = [
-    { version = "^2.1.0", source = "PyPI", platform = "darwin", optional = true },
-    { version = "^2.1.0", source = "torch-cpu", platform = "!=darwin", optional = true },
+    { version = "^2.2.0", source = "PyPI", platform = "darwin", optional = true },
+    { version = "^2.2.0", source = "torch-cpu", platform = "!=darwin", optional = true },
 ]
 
 # scikit-rf