update deps to correct scikit-learn

neuropsychology#388
birm · Nov 27, 2020 · 39dce3d · 39dce3d
1 parent c4e210e
commit 39dce3d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/Pipfile b/Pipfile
@@ -19,7 +19,7 @@ yapf = "*"
 bioread = "*"
 mne = "*"
 scipy = "*"
-sklearn = "*"
+scikit-learn = "*"
 matplotlib = "*"
 pyentrp = "*"
 cvxopt = "*"

diff --git a/docs/conf.py b/docs/conf.py
@@ -31,7 +31,7 @@
 
 # -- Mock modules ---------------------------------------------
 MOCK_MODULES = ['scipy', 'scipy.signal', 'scipy.ndimage', 'scipy.stats', 'scipy.misc', 'scipy.interpolate', 'scipy.sparse', 'scipy.linalg',
-                'scipy.spatial', 'scipy.special', 'scipy.integrate', 'scipy.cluster', 'scipy.optimize',
+                'scipy.spatial', 'scipy.special', 'scipy.integrate', 'scipy.cluster', 'scipy.optimize', "scikit-learn",
                 'sklearn', 'sklearn.neighbors', 'sklearn.mixture', 'sklearn.datasets', 'sklearn.metrics', 'sklearn.metrics.pairwise', 'sklearn.decomposition',
                 'sklearn.cluster', 'sklearn.cross_validation', 'mne', 'bioread', 'cvxopt', 'pywt']
 

diff --git a/setup.py b/setup.py
@@ -22,7 +22,7 @@ def find_version():
 
 
 # Dependencies
-requirements = ["numpy", "pandas", "scipy", "sklearn", "matplotlib"]
+requirements = ["numpy", "pandas", "scipy", "scikit-learn", "matplotlib"]
 
 # Optional Dependencies (only needed / downloaded for testing purposes, for instance to test against some other packages)
 setup_requirements = ["pytest-runner", "numpy"]