Raise AttributeError on non-exist artifact (#1971)

Netflix · Aug 20, 2024 · 2bf2a61 · 2bf2a61
1 parent 70bdff0
commit 2bf2a61
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/metaflow/client/core.py b/metaflow/client/core.py
@@ -1027,6 +1027,8 @@ def __init__(self, artifacts: Iterable[DataArtifact]):
         self._artifacts = dict((art.id, art) for art in artifacts)
 
     def __getattr__(self, name: str):
+        if name not in self._artifacts:
+            raise AttributeError(name)
         return self._artifacts[name].data
 
     def __contains__(self, var):