Update examples/preferential-optimization/evaluator.py

This commit is contained in:
contramundum53
2023-08-16 16:49:23 +09:00
committed by GitHub
parent c66de309c1
commit 0a6cb1d9ff
@@ -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")