Skip to content

Commit 262c48d

Browse files
committed
Fix
1 parent a1a5953 commit 262c48d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytorch/pytorch_checkpoint.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ def objective(trial):
9191

9292
trial_number = RetryFailedTrialCallback.retried_trial_number(trial)
9393

94-
artifact_id = trial.study.trials[trial_number].user_attrs.get("artifact_id")
94+
artifact_id = trial_number and trial.study.trials[trial_number].user_attrs.get("artifact_id")
9595
if trial_number is not None and artifact_id is not None:
9696
download_artifact(
9797
artifact_store=artifact_store,

0 commit comments

Comments
 (0)