chore: bump Pytorch version to 2.2.1 in tests (#230)

Signed-off-by: SdgJlbl <[email protected]>
Substra · Jul 25, 2024 · d91ab40 · d91ab40
1 parent 5c3d3e4
commit d91ab40
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/changes/230.fixed b/changes/230.fixed
@@ -0,0 +1 @@
+Bump pytorch version to 2.2.1 in tests.
diff --git a/tests/algorithms/pytorch/test_base_algo.py b/tests/algorithms/pytorch/test_base_algo.py
@@ -472,7 +472,7 @@ def test_gpu(
     algo_class, strategy_class, use_gpu = dummy_gpu
     my_algo = algo_class()
     algo_deps = Dependency(
-        pypi_dependencies=["torch==2.0.1", "numpy==1.24.3", "pytest"],
+        pypi_dependencies=["torch==2.2.1", "numpy==1.24.3", "pytest"],
         editable_mode=True,
     )
 

diff --git a/tests/conftest.py b/tests/conftest.py
@@ -311,7 +311,7 @@ def aggregation_node(network):
 def torch_cpu_dependency():
     return Dependency(
         pypi_dependencies=[
-            "torch==2.0.1",
+            "torch==2.2.1",
             "numpy==1.24.3",
             "--extra-index-url https://download.pytorch.org/whl/cpu",
         ],