Merge pull request #16 from AleksandarPetrov/patch-1

Merge setup.py missing requirements
MadryLab · Nov 1, 2019 · 6156558 · 6156558
2 parents 1da08f4 + e9a0e4f
commit 6156558
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -83,7 +83,10 @@
   #  your project is installed. For an analysis of "install_requires" vs pip's
   #  requirements files see:
   #  https://packaging.python.org/en/latest/requirements.html
-  install_requires=['tqdm', 'grpcio', 'psutil', 'gitpython','py3nvml'],
+  install_requires=['tqdm', 'grpcio', 'psutil', 'gitpython','py3nvml', 'cox',
+		    'scikit-learn', 'seaborn', 'torch', 'torchvision', 'pandas',
+		    'numpy', 'scipy', 'GPUtil', 'dill', 'tensorboardX', 'tables',
+		    'matplotlib'],
   test_suite='nose.collector',
   tests_require=['nose'],
 )