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