examples: Show the artifact decorator usage

Signed-off-by: Stefano Fioravanzo <stefano@arrikto.com> Reviewed-by: Ilias Katsakioris <elikatsis@arrikto.com>
kubeflow-kale · Apr 14, 2021 · 394a5b5 · 394a5b5
1 parent 78bf67d
commit 394a5b5
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/examples/sdk/artifact.py b/examples/sdk/artifact.py
@@ -0,0 +1,26 @@
+"""
+This pipeline showcases how you can create a KFP artifact as part
+of a step.
+"""
+
+from kale.sdk import pipeline, step, artifact
+
+
+# Annotate the step with the @artifact decorator and specify the path to
+# a HTML file
+@artifact(name="test-artifact", path="/home/jovyan/myartifact.html")
+@step(name="artifact_generator")
+def generate_artifact():
+    print("Creating HTML artifact...")
+    with open("/home/jovyan/myartifact.html", "w") as f:
+        f.write("<html>Hello, World!<html>")
+    print("HTML artifact created successfully!")
+
+
+@pipeline(name="generate-artifact", experiment="generate-artifact")
+def artifact_pipeline():
+    generate_artifact()
+
+
+if __name__ == "__main__":
+    artifact_pipeline()