Increase upper torch version limit for e2e test

adap · Sep 9, 2024 · 8d4d1df · 8d4d1df
1 parent 50965ec
commit 8d4d1df
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/e2e/e2e-pytorch/pyproject.toml b/e2e/e2e-pytorch/pyproject.toml
@@ -9,7 +9,7 @@ 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",
+    "torch>=1.12.0,<=2.0.1",
     "torchvision>=0.15.1,<0.16.0",
     "tqdm>=4.63.0,<5.0.0",
 ]