re-introduced torch dependency

PreferredAI · Jul 18, 2018 · 68b2690 · 68b2690
1 parent ae387af
commit 68b2690
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/requirement.txt b/requirement.txt
@@ -1 +1 @@
-http://download.pytorch.org/whl/cpu/torch-0.4.0-cp36-cp36m-linux_x86_64.whl
+http://download.pytorch.org/whl/cpu/torch-0.4.0-cp35-cp35m-linux_x86_64.whl 
diff --git a/setup.py b/setup.py
@@ -60,7 +60,7 @@
     url="",
     zip_safe=False,
     ext_modules = external_modules,
-	install_requires=['numpy', 'scipy', 'pandas'],
+	install_requires=['numpy', 'scipy', 'pandas','torch==0.4.0'],
 	dependency_links = [torch_dl],
     packages=setuptools.find_packages(),
     classifiers=(