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

2022-07-09 05:56:46 +08:00 · 2022-07-09 05:56:46 +08:00 · 701f32b227
commit 701f32b227
parent 794cbaa60a
1 changed files with 1 additions and 1 deletions
--- a/python-package/xgboost/sklearn.py
+++ b/python-package/xgboost/sklearn.py
@ -547,7 +547,7 @@ class XGBModel(XGBModelBase):
        **kwargs: Any
    ) -> None:
        if not SKLEARN_INSTALLED:
-            raise XGBoostError(
+            raise ImportError(
                "sklearn needs to be installed in order to use this module"
            )
        self.n_estimators = n_estimators