bump torch, revert to save

protectai · Jul 25, 2024 · 1d0a35b · 1d0a35b
1 parent e23bca2
commit 1d0a35b
Show file tree

Hide file tree

Showing 3 changed files with 57 additions and 127 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -21,10 +21,9 @@ h5py = { version = "^3.9.0", optional = true }
 
 # TODO: Add py3.12 once TF release supports
 tensorflow = { version = "^2.16.2", optional = true }
-tensorflow-macos = { version = "^2.16.2", platform = "darwin", optional = true }
 
 [tool.poetry.extras]
-tensorflow = ["tensorflow", "tensorflow-macos"]
+tensorflow = ["tensorflow"]
 h5py = ["h5py"]
 
 [tool.poetry.group.test.dependencies]
@@ -35,7 +34,7 @@ requests = "^2.31.0"
 aiohttp = "^3.8.5"
 dill = "^0.3.7"
 types-requests = ">1.26"
-torch = "^2.1.2"
+torch = "^2.2"
 tf-keras = "^2.16.0"