fix: bump numpy version in tests

Signed-off-by: SdgJlbl <sarah.diot-girard@owkin.com>
Substra · Oct 1, 2024 · a8e46c3 · a8e46c3
1 parent ba2eb8f
commit a8e46c3
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/benchmark/camelyon/workflows.py b/benchmark/camelyon/workflows.py
@@ -93,9 +93,9 @@ def substrafl_fed_avg(
     )
 
     pypi_dependencies = [
-        "numpy==1.26.4",
-        "torch==2.3.0",
-        "scikit-learn==1.5.1",
+        "numpy==2.1.1",
+        "torch==2.4.1",
+        "scikit-learn==1.5.2",
     ]
     if not use_gpu:
         pypi_dependencies += ["--extra-index-url https://download.pytorch.org/whl/cpu"]

diff --git a/tests/algorithms/pytorch/test_base_algo.py b/tests/algorithms/pytorch/test_base_algo.py
@@ -471,7 +471,7 @@ def test_gpu(
     algo_class, strategy_class, disable_gpu = dummy_gpu
     my_algo = algo_class()
     algo_deps = Dependency(
-        pypi_dependencies=["torch==2.2.1", "numpy==1.26.4", "pytest"],
+        pypi_dependencies=["torch==2.4.1", "numpy==2.2.1", "pytest"],
         editable_mode=True,
     )
 

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