diff --git a/setup.py b/setup.py index 2175943..dd144db 100644 --- a/setup.py +++ b/setup.py @@ -20,7 +20,7 @@ 'scipy', 'tqdm', 'h5py', - 'sklearn', + 'scikit-learn', 'tqdm', 'sphinx_rtd_theme' ]