Merge pull request #228 from gnes-ai/fix-setup-dep

fix(setup): remove sklearn from base deps
gnes-ai · Sep 6, 2019 · fa6ab54 · fa6ab54
2 parents 22efe72 + c988c7d
commit fa6ab54
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -53,8 +53,7 @@
     'grpcio',
     'ruamel.yaml>=0.15.89',
     'pyzmq>=17.1.0',
-    'aiohttp',
-    'sklearn']
+    'aiohttp']
 
 # using pip install gnes[xx] is depreciated
 # extras_dep is kept for legacy issue, will be removed soon
@@ -73,7 +72,7 @@
     'transformers': ['pytorch-transformers'],
     'onnx': ['onnxruntime'],
     'audio': ['librosa>=0.7.0'],
-    'scipy': ['scipy'],
+    'scipy': ['scipy', 'sklearn'],
     'flask': ['flask']
 }