Merge pull request optuna#5818 from boringbyte/fix/_callbacks.py

Simplify type annotations for `optuna/_callbacks.py`
HideakiImamura · Dec 3, 2024 · 1934ea7 · 1934ea7
2 parents babf820 + 16baac6
commit 1934ea7
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/optuna/_callbacks.py b/optuna/_callbacks.py
@@ -1,5 +1,6 @@
+from __future__ import annotations
+
 from typing import Container
-from typing import Optional
 
 import optuna
 from optuna.trial import FrozenTrial
@@ -45,7 +46,7 @@ def objective(trial):
     """
 
     def __init__(
-        self, n_trials: int, states: Optional[Container[TrialState]] = (TrialState.COMPLETE,)
+        self, n_trials: int, states: Container[TrialState] | None = (TrialState.COMPLETE,)
     ) -> None:
         self._n_trials = n_trials
         self._states = states