Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add optional dependencies to setup.py (#4655)

Browse Source
This commit is contained in:
Oleksandr Pryimak 2019-07-16 02:12:44 -07:00 committed by Jiaming Yuan
parent 6323ef94ad
commit 5544a730f1
2 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
python-package/setup.py
View File

@ -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,

7
python-package/setup_pip.py
View File

@ -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,