diff --git a/python-package/lightgbm/sklearn.py b/python-package/lightgbm/sklearn.py index a1301f98323f..b5515ee63855 100644 --- a/python-package/lightgbm/sklearn.py +++ b/python-package/lightgbm/sklearn.py @@ -83,7 +83,7 @@ def __init__(self, func: _LGBM_ScikitCustomObjectiveFunction): """ self.func = func - def __call__(self, preds, dataset): + def __call__(self, preds: np.ndarray, dataset: Dataset) -> Tuple[np.ndarray, np.ndarray]: """Call passed function with appropriate arguments. Parameters @@ -165,7 +165,7 @@ def __init__(self, func: _LGBM_ScikitCustomEvalFunction): """ self.func = func - def __call__(self, preds, dataset): + def __call__(self, preds: np.ndarray, dataset: Dataset) -> Tuple[str, float, bool]: """Call passed function with appropriate arguments. Parameters