diff --git a/python-package/setup.py b/python-package/setup.py index e1433af05..c2e6fd994 100644 --- a/python-package/setup.py +++ b/python-package/setup.py @@ -288,7 +288,7 @@ if __name__ == '__main__': }, extras_require={ 'pandas': ['pandas'], - 'sklearn': ['sklearn'], + 'scikit-learn': ['scikit-learn'], 'dask': ['dask', 'pandas', 'distributed'], 'datatable': ['datatable'], 'plotting': ['graphviz', 'matplotlib']