From f868932837acf681521f5474b6f7a7cee891aac5 Mon Sep 17 00:00:00 2001 From: Michael Schuster Date: Tue, 3 Dec 2024 17:24:05 +0100 Subject: [PATCH] Fix artifact uris --- src/zenml/orchestrators/output_utils.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/zenml/orchestrators/output_utils.py b/src/zenml/orchestrators/output_utils.py index 8bd18e09ab5..0d2cb400683 100644 --- a/src/zenml/orchestrators/output_utils.py +++ b/src/zenml/orchestrators/output_utils.py @@ -19,6 +19,7 @@ from zenml.client import Client from zenml.logger import get_logger +from zenml.utils import string_utils if TYPE_CHECKING: from zenml.artifact_stores import BaseArtifactStore @@ -75,10 +76,13 @@ def prepare_output_artifact_uris( artifact_store = stack.artifact_store output_artifact_uris: Dict[str, str] = {} for output_name in step.config.outputs.keys(): + substituted_output_name = string_utils.format_name_template( + output_name, substitutions=step_run.config.substitutions + ) artifact_uri = generate_artifact_uri( artifact_store=stack.artifact_store, step_run=step_run, - output_name=output_name, + output_name=substituted_output_name, ) if artifact_store.exists(artifact_uri): raise RuntimeError("Artifact already exists")