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

#5833)
microsoft · Apr 19, 2023 · f065b43 · f065b43
1 parent 5989405
commit f065b43
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python-package/lightgbm/basic.py b/python-package/lightgbm/basic.py
@@ -1736,7 +1736,7 @@ def _set_init_score_by_predictor(
         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: