Skip to content

Inner Steps

This example shows how to use a Steps template within another Steps template.

Note, this is different from parallel steps, as parallel steps are a collection of substeps to run in parallel, whereas this inner steps example shows the use of a whole Steps template (which will run sequentially in this example).

 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
31
32
from hera.workflows import Parameter, Steps, Workflow, script


@script()
def hello(name: str):
    print("Hello, {name}!".format(name=name))


with Workflow(
    generate_name="callable-inner-steps-",
    entrypoint="calling-steps",
) as w:
    with Steps(
        name="my-steps",
        inputs=Parameter(name="my-step-input"),
    ) as my_steps:
        hello(
            name="hello-1",
            arguments={
                "name": f"hello-1-{my_steps.get_parameter('my-step-input')}",
            },
        )
        hello(
            name="hello-2",
            arguments={
                "name": f"hello-2-{my_steps.get_parameter('my-step-input')}",
            },
        )

    with Steps(name="calling-steps") as s:
        my_steps(name="call-1", arguments={"my-step-input": "call-1"})
        my_steps(name="call-2", arguments={"my-step-input": "call-2"})

 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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
apiVersion: argoproj.io/v1alpha1
kind: Workflow
metadata:
  generateName: callable-inner-steps-
spec:
  entrypoint: calling-steps
  templates:
  - name: my-steps
    steps:
    - - name: hello-1
        template: hello
        arguments:
          parameters:
          - name: name
            value: hello-1-{{inputs.parameters.my-step-input}}
    - - name: hello-2
        template: hello
        arguments:
          parameters:
          - name: name
            value: hello-2-{{inputs.parameters.my-step-input}}
    inputs:
      parameters:
      - name: my-step-input
  - name: hello
    inputs:
      parameters:
      - name: name
    script:
      image: python:3.10
      source: |-
        import os
        import sys
        sys.path.append(os.getcwd())
        import json
        try: name = json.loads(r'''{{inputs.parameters.name}}''')
        except: name = r'''{{inputs.parameters.name}}'''

        print('Hello, {name}!'.format(name=name))
      command:
      - python
  - name: calling-steps
    steps:
    - - name: call-1
        template: my-steps
        arguments:
          parameters:
          - name: my-step-input
            value: call-1
    - - name: call-2
        template: my-steps
        arguments:
          parameters:
          - name: my-step-input
            value: call-2

Comments