diff --git a/benchmark/camelyon/workflows.py b/benchmark/camelyon/workflows.py index 43a6a2ef..7b1b95d1 100644 --- 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/changes/252.fixed b/changes/252.fixed new file mode 100644 index 00000000..3be200a8 --- /dev/null +++ b/changes/252.fixed @@ -0,0 +1 @@ +Bump NumPy and pytorch versions in tests. diff --git a/tests/algorithms/pytorch/test_base_algo.py b/tests/algorithms/pytorch/test_base_algo.py index 700813e1..d64d768b 100644 --- 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 index b0a3bd7e..a090bffa 100644 --- 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,