update the torch version constraint

kathyxchen · kathyxchen · commit 8edb6379e78e · 2020-02-25T15:16:13.000-05:00
diff --git a/setup.py b/setup.py
@@ -62,7 +62,5 @@
         "scipy",
         "seaborn",
         "statsmodels",
-        "torch>=0.4.1",
-        "torch<=1.4.0",
-        "torchvision"
+        "torch>=0.4.1, <=1.4.0",
     ])
