Skip to content

Template Sets

This example shows how to use the Hera concept of TemplateSet.

A TemplateSet lets you write templates that are not attached to a particular Workflow. You can then add the TemplateSet to a Workflow at submission time. This can be useful if you just want to distribute Python packages. Read more in the Best Practices guide!

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
from hera.shared import global_config
from hera.workflows import Output, TemplateSet, Workflow

global_config.experimental_features["decorator_syntax"] = True

w = Workflow(name="workflow-using-template-sets")
templates = TemplateSet()


@templates.script()
def setup() -> Output:
    return Output(result="Setting things up")


@templates.dag()
def my_dag():
    setup(name="task-a")
    setup(name="task-b")


w.add_template_set(templates)
w.set_entrypoint(my_dag)

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
apiVersion: argoproj.io/v1alpha1
kind: Workflow
metadata:
  name: workflow-using-template-sets
spec:
  entrypoint: my-dag
  templates:
  - name: setup
    script:
      image: python:3.9
      source: '{{inputs.parameters}}'
      args:
      - -m
      - hera.workflows.runner
      - -e
      - examples.workflows.experimental.template_sets:setup
      command:
      - python
      env:
      - name: hera__outputs_directory
        value: /tmp/hera-outputs
      - name: hera__script_pydantic_io
        value: ''
  - name: my-dag
    dag:
      tasks:
      - name: task-a
        template: setup
      - name: task-b
        template: setup

Comments