Skip to content

Callable Steps

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
from hera.workflows import Parameter, Steps, Workflow, script


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


with Workflow(
    generate_name="callable-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": "hello-1-{{inputs.parameters.my-step-input}}"})
        hello(name="hello-2", arguments={"name": "hello-2-{{inputs.parameters.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-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.9
      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