try to fix tests

2025-12-30 08:02:29 +00:00 · 2022-10-29 11:44:17 -07:00
parent cb6d749821
commit 59c8948c6a
1 changed files with 3 additions and 1 deletions
--- a/setup.py
+++ b/setup.py
@@ -24,7 +24,9 @@ setup(
    'pytest-runner',
  ],
  tests_require=[
-    'pytest'
+    'pytest',
+    'torch==1.12.1',
+    'torchvision==0.13.1'
  ],
  classifiers=[
    'Development Status :: 4 - Beta',