Skip to content

Commit 5303878

Browse files
committed
CI fixes
1 parent 87e2a54 commit 5303878

File tree

2 files changed

+8
-2
lines changed

2 files changed

+8
-2
lines changed

src/zenml/execution/pipeline/dynamic/outputs.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -140,11 +140,17 @@ def get_artifact(self, key: str) -> ArtifactFuture:
140140
Args:
141141
key: The key of the artifact future.
142142
143+
Raises:
144+
KeyError: If no artifact for the given name exists.
145+
143146
Returns:
144147
The artifact future.
145148
"""
146149
if key not in self._output_keys:
147-
raise KeyError(f"Invalid key: {key}")
150+
raise KeyError(
151+
f"Step run {self._invocation_id} does not have an output with "
152+
f"the name: {key}."
153+
)
148154

149155
return ArtifactFuture(
150156
wrapped=self._wrapped,

src/zenml/models/v2/core/artifact_version.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -455,7 +455,7 @@ def load(self, disable_cache: bool = False) -> Any:
455455
cache = InMemoryArtifactCache.get()
456456

457457
if cache and (data := cache.get_artifact_data(self.id)):
458-
logger.debug(f"Returning artifact data (%s) from cache", self.id)
458+
logger.debug("Returning artifact data (%s) from cache", self.id)
459459
return data
460460

461461
data = load_artifact_from_response(self)

0 commit comments

Comments
 (0)