support light-the-torch>=0.3.3 (#15)

pmeier · Aug 23, 2021 · b5afb07 · b5afb07
1 parent a287b78
commit b5afb07
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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
@@ -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),
     )