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

[py-sckl] Raise import error if skl is not installed. (#8049)

Browse Source
This commit is contained in:
Jiaming Yuan 2022-07-09 05:56:46 +08:00 committed by GitHub
parent 794cbaa60a
commit 701f32b227
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python-package/xgboost/sklearn.py
View File

@ -547,7 +547,7 @@ class XGBModel(XGBModelBase):
**kwargs: Any **kwargs: Any
) -> None: ) -> None:
if not SKLEARN_INSTALLED: if not SKLEARN_INSTALLED:
raise XGBoostError( raise ImportError(
"sklearn needs to be installed in order to use this module" "sklearn needs to be installed in order to use this module"
) )
self.n_estimators = n_estimators self.n_estimators = n_estimators