[ci] [python-package] fix mypy errors in Booster.refit() (#5853)

ef5acfb4 · James Lamb · GitHub · 5b72e4ab · ef5acfb4
Unverified Commit ef5acfb4 authored Apr 25, 2023 by James Lamb Committed by GitHub Apr 25, 2023
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

python-package/lightgbm/basic.py python-package/lightgbm/basic.py +1 -1

No files found.
--- a/python-package/lightgbm/basic.py
+++ b/python-package/lightgbm/basic.py
@@ -4189,7 +4189,7 @@ class Booster:
        if dataset_params is None:
            dataset_params = {}
        predictor = self._to_predictor(pred_parameter=deepcopy(kwargs))
-        leaf_preds = predictor.predict(
+        leaf_preds: np.ndarray = predictor.predict(  # type: ignore[assignment]
            data=data,
            start_iteration=-1,
            pred_leaf=True,