From b5afb073e03fa7980eb095f83381d9f4ce1a8aad Mon Sep 17 00:00:00 2001 From: Philip Meier Date: Mon, 23 Aug 2021 15:36:09 +0200 Subject: [PATCH] support `light-the-torch>=0.3.3` (#15) --- setup.cfg | 2 +- tests/test_plugin.py | 4 ++-- tox_ltt/plugin.py | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/setup.cfg b/setup.cfg index 003501c..4159008 100644 --- a/setup.cfg +++ b/setup.cfg @@ -25,7 +25,7 @@ packages = find: include_package_data = True python_requires = >=3.6 install_requires = - light-the-torch >= 0.3, < 0.3.2 + light-the-torch >= 0.3.3 tox [options.packages.find] diff --git a/tests/test_plugin.py b/tests/test_plugin.py index 73b7e02..0b85866 100644 --- a/tests/test_plugin.py +++ b/tests/test_plugin.py @@ -198,7 +198,7 @@ def test_tox_ltt_pytorch_force_cpu( result.assert_success(is_run_test_env=False) _, kwargs = mock.call_args - assert kwargs["computation_backend"] == CPUBackend() + assert kwargs["computation_backends"] == CPUBackend() @pytest.mark.slow @@ -213,7 +213,7 @@ def test_tox_ltt_force_cpu_legacy( result.assert_success(is_run_test_env=False) _, kwargs = mock.call_args - assert kwargs["computation_backend"] == CPUBackend() + assert kwargs["computation_backends"] == CPUBackend() def test_tox_ltt_no_requirements( diff --git a/tox_ltt/plugin.py b/tox_ltt/plugin.py index c8b6ecd..ef235fc 100644 --- a/tox_ltt/plugin.py +++ b/tox_ltt/plugin.py @@ -102,7 +102,7 @@ def tox_testenv_install_deps(venv: VirtualEnv, action: Action) -> None: links = ltt.find_links( dists, - computation_backend=get_computation_backend(envconfig), + computation_backends=get_computation_backend(envconfig), channel=envconfig.pytorch_channel, python_version=get_python_version(envconfig), )