diff --git a/setup.py b/setup.py index 595081a..1bb85a4 100644 --- a/setup.py +++ b/setup.py @@ -26,7 +26,7 @@ install_requires=[ 'numpy', 'pandas', - 'sklearn', + 'scikit-learn', 'joblib' ], test_suite='nose.collector',