diff --git a/changes/230.fixed b/changes/230.fixed new file mode 100644 index 00000000..01cf4583 --- /dev/null +++ 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 index c35f3e33..48893d80 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.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 index d540d4a2..287f0e38 100644 --- 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", ],