[python-package] fix mypy errors about custom eval functions in sklea…

…rn.py (#5721)
microsoft · Feb 16, 2023 · f136de4 · f136de4
1 parent 74316c1
commit f136de4
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/python-package/lightgbm/sklearn.py b/python-package/lightgbm/sklearn.py
@@ -178,7 +178,11 @@ def __init__(self, func: _LGBM_ScikitCustomEvalFunction):
         """
         self.func = func
 
-    def __call__(self, preds: np.ndarray, dataset: Dataset) -> Tuple[str, float, bool]:
+    def __call__(
+        self,
+        preds: np.ndarray,
+        dataset: Dataset
+    ) -> Union[_LGBM_EvalFunctionResultType, List[_LGBM_EvalFunctionResultType]]:
         """Call passed function with appropriate arguments.
 
         Parameters