Add sklearn to install_requires, update install docs

TeamHG-Memex · Jan 13, 2017 · 645c3ea · 645c3ea
1 parent 417358f
commit 645c3ea
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/docs/source/overview.rst b/docs/source/overview.rst
@@ -5,10 +5,8 @@ Installation
 ------------
 
 ELI5 works in Python 2.7 and Python 3.4+. Currently it requires
-scikit-learn 0.18+, so make sure scikit-learn is installed first,
-then install eli5 using pip::
+scikit-learn 0.18+. You can install ELI5 using pip::
 
-    pip install 'scikit-learn > 0.18'
     pip install eli5
 
 Features

diff --git a/setup.py b/setup.py
@@ -37,6 +37,7 @@ def get_long_description():
         'scipy',
         'singledispatch >= 3.4.0.3',
         'six',
+        'scikit-learn >= 0.18',
         'typing',
         'graphviz',
         'tabulate>=0.7.7',