Add ArtifactLoaders to new_dag_decorator_artifacts

Signed-off-by: Elliot Gunton <[email protected]>
argoproj-labs · Sep 30, 2024 · 7a79a66 · 7a79a66
1 parent 24d6ff6
commit 7a79a66
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/docs/examples/workflows/experimental/new_dag_decorator_artifacts.md b/docs/examples/workflows/experimental/new_dag_decorator_artifacts.md
@@ -11,7 +11,7 @@
     from typing_extensions import Annotated
 
     from hera.shared import global_config
-    from hera.workflows import Artifact, Input, Output, Workflow
+    from hera.workflows import Artifact, ArtifactLoader, Input, Output, Workflow
 
     global_config.experimental_features["decorator_syntax"] = True
 
@@ -24,8 +24,8 @@
 
 
     class ConcatInput(Input):
-        word_a: Annotated[str, Artifact(name="word_a")]
-        word_b: Annotated[str, Artifact(name="word_b")]
+        word_a: Annotated[str, Artifact(name="word_a", loader=ArtifactLoader.json)]
+        word_b: Annotated[str, Artifact(name="word_b", loader=ArtifactLoader.json)]
 
 
     @w.script()

diff --git a/examples/workflows/experimental/new_dag_decorator_artifacts.py b/examples/workflows/experimental/new_dag_decorator_artifacts.py
@@ -1,7 +1,7 @@
 from typing_extensions import Annotated
 
 from hera.shared import global_config
-from hera.workflows import Artifact, Input, Output, Workflow
+from hera.workflows import Artifact, ArtifactLoader, Input, Output, Workflow
 
 global_config.experimental_features["decorator_syntax"] = True
 
@@ -14,8 +14,8 @@ class ArtifactOutput(Output):
 
 
 class ConcatInput(Input):
-    word_a: Annotated[str, Artifact(name="word_a")]
-    word_b: Annotated[str, Artifact(name="word_b")]
+    word_a: Annotated[str, Artifact(name="word_a", loader=ArtifactLoader.json)]
+    word_b: Annotated[str, Artifact(name="word_b", loader=ArtifactLoader.json)]
 
 
 @w.script()