Clean installation requirements in setup.py

2020-04-06 15:58:35 +02:00 · 2020-04-06 15:58:35 +02:00 · f9bc4a29c9
commit f9bc4a29c9
parent a113b45a73
1 changed files with 4 additions and 8 deletions
--- a/setup.py
+++ b/setup.py
@ -25,17 +25,13 @@ setup(name='prototorch',
          'torch>=1.3.1',
          'torchvision>=0.5.0',
          'numpy>=1.9.1',
-          'matplotlib',
-          'sklearn',
      ],
      extras_require={
-          'other': [
-              'xlrd',
-              'pandas',
-              'seaborn',
-              'imageio',
+          'examples': [
+              'sklearn',
+              'matplotlib',
          ],
-          'tests': ['pytest'],
+          'tests': ['unittest'],
      },
      classifiers=[
          'Development Status :: 2 - Pre-Alpha', 'Environment :: Console',