Skip to content

GitLab

Unverified Commit 611ce0d9 authored by James Lamb's avatar James Lamb Committed by GitHub
Browse files

[ci] [python-package] fix mypy error about dask._predict_part() (#5849)

parent ac37bf8a
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
python-package/lightgbm/dask.py
View file @ 611ce0d9
...@@ -836,6 +836,7 @@ def _predict_part( ...@@ -836,6 +836,7 @@ def _predict_part(
**kwargs: Any **kwargs: Any
) -> _DaskPart: ) -> _DaskPart:
result: _DaskPart
if part.shape[0] == 0: if part.shape[0] == 0:
result = np.array([]) result = np.array([])
elif pred_proba: elif pred_proba:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment