Skip to content

GitLab

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
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
python-package/lightgbm/basic.py
View file @ f065b436
...@@ -1736,7 +1736,7 @@ class Dataset: ...@@ -1736,7 +1736,7 @@ class Dataset:
self, self,
predictor: Optional[_InnerPredictor], predictor: Optional[_InnerPredictor],
data: _LGBM_TrainDataType, data: _LGBM_TrainDataType,
used_indices: Optional[List[int]] used_indices: Optional[Union[List[int], np.ndarray]]
) -> "Dataset": ) -> "Dataset":
data_has_header = False data_has_header = False
if isinstance(data, (str, Path)) and self.params is not None: 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