diff --git a/setup.py b/setup.py index da7649b8..ad6e8c92 100755 --- a/setup.py +++ b/setup.py @@ -51,7 +51,7 @@ 'simplejson', 'pyyaml', # YAML reader 'joblib', # Parallelism for dummies - 'sklearn', + 'scikit-learn', 'scikit-optimize', ],