From f84291c1e1b973b5c752f9a5678c022defa4a67b Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Sun, 23 Jan 2022 16:33:23 +0800 Subject: [PATCH] Fix `max_cat_to_onehot` doc annotation [skip ci] (#7592) --- 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 374958f750c4..d6ab64ef2917 100644 --- a/python-package/xgboost/sklearn.py +++ b/python-package/xgboost/sklearn.py @@ -197,7 +197,7 @@ def inner(y_score: np.ndarray, dmatrix: DMatrix) -> Tuple[str, float]: Experimental support for categorical data. Do not set to true unless you are interested in development. Only valid when `gpu_hist` and dataframe are used. - max_cat_to_onehot : bool + max_cat_to_onehot : Optional[int] .. versionadded:: 1.6.0