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

[python-package] fix mypy error about used_indices in Dataset.subset() (#5833)

parent 5989405c
......@@ -1736,7 +1736,7 @@ class Dataset:
self,
predictor: Optional[_InnerPredictor],
data: _LGBM_TrainDataType,
used_indices: Optional[List[int]]
used_indices: Optional[Union[List[int], np.ndarray]]
) -> "Dataset":
data_has_header = False
if isinstance(data, (str, Path)) and self.params is not None:
......
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