Merge pull request #192 from inferno-pytorch/fix-setup-requirements

fix setup.py requirements
inferno-pytorch · Sep 2, 2019 · 76773dd · 76773dd
2 parents 546c395 + bcb2b3d
commit 76773dd
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -23,7 +23,9 @@
     "scipy>=0.13.0",
     "h5py",
     "numpy>=1.8",
-    "scikit-image"
+    "scikit-image",
+    "torchvision",
+    "tqdm"
 ]