Fix deps

adap · Nov 6, 2024 · 24813a6 · 24813a6
1 parent 0b2a3cf
commit 24813a6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/e2e/e2e-pytorch/pyproject.toml b/e2e/e2e-pytorch/pyproject.toml
@@ -9,8 +9,8 @@ description = "PyTorch Federated Learning E2E test with Flower"
 license = "Apache-2.0"
 dependencies = [
     "flwr[simulation] @ {root:parent:parent:uri}",
-    "torch>=1.12.0,<2.0.0",
-    "torchvision>=0.14.1,<0.15.0",
+    "torch>=2.5.0,<3.0.0",
+    "torchvision>=0.20.1,<0.21.0",
     "tqdm>=4.63.0,<5.0.0",
 ]
 

diff --git a/e2e/e2e-scikit-learn/pyproject.toml b/e2e/e2e-scikit-learn/pyproject.toml
@@ -15,6 +15,7 @@ dependencies = [
     "flwr[simulation,rest] @ {root:parent:parent:uri}",
     "scikit-learn>=1.1.1,<2.0.0",
     "openml>=0.14.0,<0.15.0",
+    "numpy<2.0.0",
 ]
 
 [tool.hatch.build.targets.wheel]