From 24813a6b76984efc3640773c1ce7dac2b00856c2 Mon Sep 17 00:00:00 2001
From: Chong Shen Ng <chong.shen@flower.ai>
Date: Wed, 6 Nov 2024 12:09:49 +0000
Subject: [PATCH] Fix deps

---
 e2e/e2e-pytorch/pyproject.toml      | 4 ++--
 e2e/e2e-scikit-learn/pyproject.toml | 1 +
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/e2e/e2e-pytorch/pyproject.toml b/e2e/e2e-pytorch/pyproject.toml
index 0e48334693d3..9e2029aecefb 100644
--- 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
index aef9a4a8a00b..03f5540aa15d 100644
--- 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]