From b94577d05cd08a6a9147bff16395d442866cb511 Mon Sep 17 00:00:00 2001 From: James Lamb Date: Mon, 29 Nov 2021 23:37:20 -0600 Subject: [PATCH] [python-package] fix mypy error about missing type hint in dask.py --- python-package/lightgbm/dask.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python-package/lightgbm/dask.py b/python-package/lightgbm/dask.py index 247accb481d4..1b112320832f 100644 --- a/python-package/lightgbm/dask.py +++ b/python-package/lightgbm/dask.py @@ -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