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

Author: pmeier

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]

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(

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),
     )