try to fix tests

p1x31 · Oct 29, 2022 · 59c8948 · 59c8948
1 parent cb6d749
commit 59c8948
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -24,7 +24,9 @@
     'pytest-runner',
   ],
   tests_require=[
-    'pytest'
+    'pytest',
+    'torch==1.12.1',
+    'torchvision==0.13.1'
   ],
   classifiers=[
     'Development Status :: 4 - Beta',