Skip to content

Commit

Permalink
[python-package] fix mypy error about missing type hint in dask.py
Browse files Browse the repository at this point in the history
  • Loading branch information
jameslamb committed Nov 30, 2021
1 parent 67b4205 commit b94577d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python-package/lightgbm/dask.py
Original file line number Diff line number Diff line change
Expand Up @@ -930,7 +930,7 @@ def _extract(items: List[Any], i: int) -> Any:
num_cols = model.n_features_ + 1

nrows_per_chunk = data.chunks[0]
out = [[] for _ in range(num_classes)]
out: List[dask_Array] = [[] for _ in range(num_classes)]

# need to tell Dask the expected type and shape of individual preds
pred_meta = data._meta
Expand Down

0 comments on commit b94577d

Please sign in to comment.