Skip to content

Commit cad5190

Browse files
authored
Merge pull request optuna#6143 from Shubham05122002/refactor-return-tuple-style-models-py
Refactor return statement style in `optuna/storages/_rdb/models.py` for consistency among the codebase
2 parents b0f1482 + 42daf94 commit cad5190

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

optuna/storages/_rdb/models.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -421,11 +421,11 @@ class TrialValueType(enum.Enum):
421421
@classmethod
422422
def value_to_stored_repr(cls, value: float) -> tuple[float | None, TrialValueType]:
423423
if value == float("inf"):
424-
return (None, cls.TrialValueType.INF_POS)
424+
return None, cls.TrialValueType.INF_POS
425425
elif value == float("-inf"):
426-
return (None, cls.TrialValueType.INF_NEG)
426+
return None, cls.TrialValueType.INF_NEG
427427
else:
428-
return (value, cls.TrialValueType.FINITE)
428+
return value, cls.TrialValueType.FINITE
429429

430430
@classmethod
431431
def stored_repr_to_value(cls, value: float | None, float_type: TrialValueType) -> float:
@@ -486,13 +486,13 @@ def intermediate_value_to_stored_repr(
486486
cls, value: float
487487
) -> tuple[float | None, TrialIntermediateValueType]:
488488
if math.isnan(value):
489-
return (None, cls.TrialIntermediateValueType.NAN)
489+
return None, cls.TrialIntermediateValueType.NAN
490490
elif value == float("inf"):
491-
return (None, cls.TrialIntermediateValueType.INF_POS)
491+
return None, cls.TrialIntermediateValueType.INF_POS
492492
elif value == float("-inf"):
493-
return (None, cls.TrialIntermediateValueType.INF_NEG)
493+
return None, cls.TrialIntermediateValueType.INF_NEG
494494
else:
495-
return (value, cls.TrialIntermediateValueType.FINITE)
495+
return value, cls.TrialIntermediateValueType.FINITE
496496

497497
@classmethod
498498
def stored_repr_to_intermediate_value(

0 commit comments

Comments
 (0)