From 701f32b22753f576ab108521e444288fd8953f54 Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Sat, 9 Jul 2022 05:56:46 +0800 Subject: [PATCH] [py-sckl] Raise import error if skl is not installed. (#8049) --- python-package/xgboost/sklearn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python-package/xgboost/sklearn.py b/python-package/xgboost/sklearn.py index 9e677a000..b9c26f6dd 100644 --- 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