Skip to content

Commit

Permalink
fix type annotations for 'optuna/storages/_rdb/alembic/versions/v3.0.…
Browse files Browse the repository at this point in the history
…0.d.py'
  • Loading branch information
boringbyte committed Dec 25, 2024
1 parent 3aa0384 commit 59f36b4
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 59f36b4

Please sign in to comment.