diff --git a/python-package/setup.py b/python-package/setup.py index 30e0d070f..739324b08 100644 --- a/python-package/setup.py +++ b/python-package/setup.py @@ -43,6 +43,13 @@ setup(name='xgboost', 'numpy', 'scipy', ], + extras_require={ + 'pandas': ['pandas'], + 'sklearn': ['sklearn'], + 'dask': ['dask', 'pandas', 'distributed'], + 'datatable': ['datatable'], + 'plotting': ['graphviz', 'matplotlib'] + }, maintainer='Hyunsu Cho', maintainer_email='chohyu01@cs.washington.edu', zip_safe=False, diff --git a/python-package/setup_pip.py b/python-package/setup_pip.py index 785429898..86ba0ae4f 100644 --- a/python-package/setup_pip.py +++ b/python-package/setup_pip.py @@ -48,6 +48,13 @@ setup(name='xgboost', 'numpy', 'scipy', ], + extras_require={ + 'pandas': ['pandas'], + 'sklearn': ['sklearn'], + 'dask': ['dask', 'pandas', 'distributed'], + 'datatable': ['datatable'], + 'plotting': ['graphviz', 'matplotlib'] + }, maintainer='Hyunsu Cho', maintainer_email='chohyu01@cs.washington.edu', zip_safe=False,