diff --git a/ci/test.py b/ci/test.py index 7772e4321d4974e699280ae419924b7874d2589d..ab41980d6b6bae18e8bf990f162b20b57edd82af 100755 --- a/ci/test.py +++ b/ci/test.py @@ -37,7 +37,6 @@ if __name__ == "__main__": ] ) - check_call([python_venv_executable(), "-m", "pip", "install", "torch"]) check_call( [ python_venv_executable(), diff --git a/setup.py b/setup.py index 19fae52e44b1e25b23f29a044016b79eafa457cd..14d44d9b7e322e78d8823e2266e77b06d8e2c12e 100644 --- a/setup.py +++ b/setup.py @@ -59,6 +59,7 @@ setup( "deprecation", "tqdm", "pre-commit", + "torch", ], classifiers=[ "Development Status :: 3 - Alpha",