From b79c9dd6fcc68d997f6ac63362de50176cf986a9 Mon Sep 17 00:00:00 2001 From: SdgJlbl Date: Thu, 25 Jul 2024 15:26:42 +0200 Subject: [PATCH] chore: bump NumPy Signed-off-by: SdgJlbl --- changes/231.fixed | 1 + tests/algorithms/pytorch/test_base_algo.py | 2 +- tests/conftest.py | 2 +- 3 files changed, 3 insertions(+), 2 deletions(-) create mode 100644 changes/231.fixed diff --git a/changes/231.fixed b/changes/231.fixed new file mode 100644 index 00000000..63c269f7 --- /dev/null +++ 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 index 48893d80..eb4332ef 100644 --- 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 index 287f0e38..b0a3bd7e 100644 --- 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,