Fix deps in e2e-scikit-learn

adap · Nov 6, 2024 · 4879170 · 4879170
1 parent a38443c
commit 4879170
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/e2e/e2e-pytorch-lightning/pyproject.toml b/e2e/e2e-pytorch-lightning/pyproject.toml
@@ -9,8 +9,8 @@ description = "Federated Learning E2E test with Flower and PyTorch Lightning"
 license = "Apache-2.0"
 dependencies = [
     "flwr[simulation] @ {root:parent:parent:uri}",
-    "pytorch-lightning==2.2.4",
-    "torchvision==0.14.1",
+    "pytorch-lightning==2.4.0",
+    "torchvision>=0.20.1,<0.21.0",
 ]
 
 [tool.hatch.build.targets.wheel]

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

diff --git a/pyproject.toml b/pyproject.toml
@@ -64,7 +64,7 @@ flower-client-app = "flwr.client.supernode:run_client_app" # Deprecated
 [tool.poetry.dependencies]
 python = "^3.9"
 # Mandatory dependencies
-numpy = "^2.0.0"
+numpy = ">=1.0.0,<3.0.0"
 grpcio = "^1.60.0,!=1.64.2,!=1.65.1,!=1.65.2,!=1.65.4,!=1.65.5,!=1.66.0,!=1.66.1"
 protobuf = "^4.25.2"
 cryptography = "^42.0.4"