diff --git a/examples/preferential-optimization/evaluator.py b/examples/preferential-optimization/evaluator.py index 71a1df9d..355af4d7 100644 --- a/examples/preferential-optimization/evaluator.py +++ b/examples/preferential-optimization/evaluator.py @@ -70,8 +70,8 @@ def main() -> NoReturn: trial = finished_dict[trial_number] col = cols[col_i] - rgb_artifact_id = trial.user_attrs.get("rgb_artifact_id") - image_caption = trial.user_attrs.get("image_caption") + rgb_artifact_id = trial.user_attrs["rgb_artifact_id"] + image_caption = trial.user_attrs["image_caption"] with col: with artifact_backend.open(rgb_artifact_id) as fsrc: tmp_img_path = os.path.join(tmpdir, rgb_artifact_id + ".png")