Skip to content

Commit 2bbd267

Browse files
vizier-teamcopybara-github
vizier-team
authored andcommitted
Pytype fix.
PiperOrigin-RevId: 722934690
1 parent 7ce0ae8 commit 2bbd267

File tree

2 files changed

+9
-3
lines changed

2 files changed

+9
-3
lines changed

vizier/_src/pyglove/converters.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -559,11 +559,13 @@ def to_tuner_trial(self, vizier_trial: vz.Trial) -> pg.tuning.Trial:
559559
vizier_trial.final_measurement
560560
),
561561
status=self._to_tuner_trial_status(vizier_trial.status),
562-
created_time=int(
562+
created_time=int( # pylint: disable=g-long-ternary
563563
vizier_trial.creation_time.replace(
564564
tzinfo=datetime.timezone.utc
565565
).timestamp()
566-
),
566+
)
567+
if vizier_trial.creation_time
568+
else None,
567569
completed_time=int( # pylint: disable=g-long-ternary
568570
vizier_trial.completion_time.replace(
569571
tzinfo=datetime.timezone.utc

vizier/_src/pyglove/core.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,10 @@ def __init__(
5959
completed_time = (
6060
int(trial.completion_time.timestamp()) if trial.completion_time else 0
6161
)
62+
created_time = (
63+
int(trial.creation_time.timestamp()) if trial.creation_time else 0
64+
)
65+
6266
super().__init__(
6367
dna=pg.DNA(None),
6468
id=trial.id,
@@ -67,7 +71,7 @@ def __init__(
6771
trial.final_measurement
6872
),
6973
status=_trial_status_legacy_value(trial.status),
70-
created_time=int(trial.creation_time.timestamp()),
74+
created_time=created_time,
7175
completed_time=completed_time,
7276
infeasible=trial.infeasible,
7377
**kwargs,

0 commit comments

Comments
 (0)