diff --git a/setup.py b/setup.py index 459852d..225abee 100644 --- a/setup.py +++ b/setup.py @@ -11,7 +11,7 @@ 'matplotlib', 'numpy', 'tqdm', - 'sklearn', + 'scikit-learn', #sklearn has been deprecated 'scipy', 'toolz', 'bayesian-optimization >= 1.1.0'