From d91ab40f5123eaa9eebb232a1735d2fbe847d5b5 Mon Sep 17 00:00:00 2001 From: SdgJlbl Date: Thu, 25 Jul 2024 10:26:10 +0200 Subject: [PATCH] chore: bump Pytorch version to 2.2.1 in tests (#230) Signed-off-by: SdgJlbl --- changes/230.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/230.fixed 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", ],