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

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

parent 74316c11
...@@ -178,7 +178,11 @@ class _EvalFunctionWrapper: ...@@ -178,7 +178,11 @@ class _EvalFunctionWrapper:
""" """
self.func = func 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. """Call passed function with appropriate arguments.
Parameters Parameters
......
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