Merge pull request #5167 from Alnusjaponica/fix-checks-integration

Fix `Checks(integration)` failure
optuna · Dec 26, 2023 · a8e6ead · a8e6ead
2 parents 7a30780 + c4cf2f5
commit a8e6ead
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/optuna/integration/lightgbm.py b/optuna/integration/lightgbm.py
@@ -10,6 +10,7 @@
 
 if TYPE_CHECKING:
     from lightgbm.basic import _LGBM_BoosterEvalMethodResultType
+    from lightgbm.basic import _LGBM_BoosterEvalMethodResultWithStandardDeviationType
     from lightgbm.callback import CallbackEnv
 
 
@@ -94,7 +95,11 @@ def __init__(
 
     def _find_evaluation_result(
         self, target_valid_name: str, env: CallbackEnv
-    ) -> _LGBM_BoosterEvalMethodResultType | None:
+    ) -> (
+        _LGBM_BoosterEvalMethodResultType
+        | _LGBM_BoosterEvalMethodResultWithStandardDeviationType
+        | None
+    ):
         evaluation_result_list = env.evaluation_result_list
         if evaluation_result_list is None:
             return None