Added sklearn as required dependency for interpret-core.

IFV · Sep 17, 2019 · 5b257c5 · 5b257c5
1 parent e318e7b
commit 5b257c5
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/python/interpret-core/setup.py b/python/interpret-core/setup.py
@@ -39,7 +39,7 @@
 sklearn_dep = "scikit-learn>=0.20.0"
 extras = {
     # Core
-    "required": ["numpy>=1.15.1", "scipy>=1.2.1", "pandas>=0.24.0"],
+    "required": ["numpy>=1.15.1", "scipy>=1.2.1", "pandas>=0.24.0", sklearn_dep],
     "debug": ["psutil>=5.6.2"],
     "notebook": ["ipykernel>=5.1.0", "ipython>=7.4.0"],
     # Plotly (required if .visualize is ever called)
@@ -48,9 +48,9 @@
     "lime": ["lime>=0.1.1.33"],
     "sensitivity": ["SALib>=1.3.3"],
     "shap": ["shap>=0.28.5", "dill"],
-    "ebm": [sklearn_dep],
-    "linear": [sklearn_dep],
-    "decisiontree": [sklearn_dep],
+    "ebm": [],
+    "linear": [],
+    "decisiontree": [],
     "skoperules": ["skope-rules>=1.0.0"],
     "treeinterpreter": ["treeinterpreter>=0.2.2"],
     # Dash