from hera.workflows import DAG, Workflow, script
@script()
def flip():
import random
result = "heads" if random.randint(0, 1) == 0 else "tails"
print(result)
@script()
def heads():
print("it was heads")
@script()
def tails():
print("it was tails")
with Workflow(generate_name="coinflip-", entrypoint="d") as w:
with DAG(name="d") as s:
f = flip()
heads().on_other_result(f, "heads")
tails().on_other_result(f, "tails")