diff --git a/optuna/storages/_rdb/alembic/versions/v3.0.0.d.py b/optuna/storages/_rdb/alembic/versions/v3.0.0.d.py index f35f2635f35..2b6a3588e50 100644 --- a/optuna/storages/_rdb/alembic/versions/v3.0.0.d.py +++ b/optuna/storages/_rdb/alembic/versions/v3.0.0.d.py @@ -6,6 +6,8 @@ """ +from __future__ import annotations + import enum import numpy as np @@ -13,8 +15,6 @@ import sqlalchemy as sa from sqlalchemy.exc import SQLAlchemyError from sqlalchemy import orm -from typing import Optional -from typing import Tuple try: from sqlalchemy.orm import declarative_base @@ -53,7 +53,7 @@ class TrialValueType(enum.Enum): def value_to_stored_repr( cls, value: float, - ) -> Tuple[Optional[float], TrialValueType]: + ) -> tuple[float | None, TrialValueType]: if value == float("inf"): return (None, cls.TrialValueType.INF_POS) elif value == float("-inf"): @@ -62,7 +62,7 @@ def value_to_stored_repr( return (value, cls.TrialValueType.FINITE) @classmethod - def stored_repr_to_value(cls, value: Optional[float], float_type: TrialValueType) -> float: + def stored_repr_to_value(cls, value: float | None, float_type: TrialValueType) -> float: if float_type == cls.TrialValueType.INF_POS: assert value is None return float("inf")