chore: bump NumPy

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

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/changes/231.fixed b/changes/231.fixed
@@ -0,0 +1 @@
+Bump NumPy version to 1.26.4 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.2.1", "numpy==1.24.3", "pytest"],
+        pypi_dependencies=["torch==2.2.1", "numpy==1.26.4", "pytest"],
         editable_mode=True,
     )
 

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