remove sklearn from base dep

gnes-ai · Sep 6, 2019 · c988c7d · c988c7d
1 parent 22efe72
commit c988c7d
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']
 }