[python-package] fix mypy errors about custom eval functions in sklearn.py (#5721)

f136de41 · James Lamb · GitHub · 74316c11 · f136de41
Unverified Commit f136de41 authored Feb 16, 2023 by James Lamb Committed by GitHub Feb 16, 2023
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

python-package/lightgbm/sklearn.py python-package/lightgbm/sklearn.py +5 -1

No files found.
--- a/python-package/lightgbm/sklearn.py
+++ b/python-package/lightgbm/sklearn.py
@@ -178,7 +178,11 @@ class _EvalFunctionWrapper:
        """
        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