[python] add type hints on sklearn metric and eval wrappers (#5238)

microsoft · May 25, 2022 · 23da5fc · 23da5fc
1 parent 1a3afd2
commit 23da5fc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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