Skip to content

With Sequence

This example showcases how to generate and parallelize generated sequences

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
from hera.workflows import DAG, Workflow, script
from hera.workflows.models import Sequence


@script()
def gen_num():
    print(3)


@script()
def say(message: str):
    print(message)


with Workflow(generate_name="with-sequence-example", entrypoint="d") as w:
    with DAG(name="d"):
        t1 = gen_num(name="t1")
        t2 = say(name="t2", with_sequence=Sequence(count=t1.result, start="0"), arguments={"message": "{{item}}"})
        t3 = say(
            name="t3",
            with_sequence=Sequence(start=t1.result, end="5", format="2020-05-%02X"),
            arguments={"message": "{{item}}"},
        )
        t1 >> [t2, t3]

 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
56
57
58
59
60
apiVersion: argoproj.io/v1alpha1
kind: Workflow
metadata:
  generateName: with-sequence-example
spec:
  entrypoint: d
  templates:
  - name: d
    dag:
      tasks:
      - name: t1
        template: gen-num
      - name: t2
        depends: t1
        template: say
        arguments:
          parameters:
          - name: message
            value: '{{item}}'
        withSequence:
          count: '{{tasks.t1.outputs.result}}'
          start: '0'
      - name: t3
        depends: t1
        template: say
        arguments:
          parameters:
          - name: message
            value: '{{item}}'
        withSequence:
          end: '5'
          format: 2020-05-%02X
          start: '{{tasks.t1.outputs.result}}'
  - name: gen-num
    script:
      image: python:3.9
      source: |-
        import os
        import sys
        sys.path.append(os.getcwd())
        print(3)
      command:
      - python
  - name: say
    inputs:
      parameters:
      - name: message
    script:
      image: python:3.9
      source: |-
        import os
        import sys
        sys.path.append(os.getcwd())
        import json
        try: message = json.loads(r'''{{inputs.parameters.message}}''')
        except: message = r'''{{inputs.parameters.message}}'''

        print(message)
      command:
      - python

Comments