test: try to make CI work with torch

carrascomj · carrascomj · commit 1efe7bf3f2f3 · 2020-03-17T22:14:06.000+01:00
diff --git a/requirements.txt b/requirements.txt
@@ -7,5 +7,4 @@ isort
 pytest
 pytest-cov
 torch
-torchvision
 wget
diff --git a/setup.py b/setup.py
@@ -10,7 +10,6 @@
         "sklearn",
         "scipy",
         "torch",
-        "torchvision",
         "wget",
     ],
     version="0.1.0",
