Skip to content

Commit

Permalink
Merge pull request optuna#5886 from boringbyte/fix/v3.0.0.d.py
Browse files Browse the repository at this point in the history
Simplify type annotations for `optuna/storages/_rdb/alembic/versions/v3.0.0.d.py`
  • Loading branch information
nabenabe0928 authored Dec 25, 2024
2 parents 21eca54 + 59f36b4 commit b93bf1b
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions optuna/storages/_rdb/alembic/versions/v3.0.0.d.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,15 +6,15 @@
"""

from __future__ import annotations

import enum

import numpy as np
from alembic import op
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
Expand Down Expand Up @@ -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"):
Expand All @@ -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")
Expand Down

0 comments on commit b93bf1b

Please sign in to comment.