Robin Kahlow
06/12/2023, 12:54 PMfrom flytekit import dynamic, task, workflow
class Something:
n: int
@task
def tk_inner() -> Something:
s = Something()
s.n = 3
return s
@task
def tk_agg(s: list[Something]) -> int:
return sum(x.n for x in s)
@dynamic
def tk_outer() -> int:
s = [tk_inner() for _ in range(5)]
return tk_agg(s=s)
@workflow
def wf_outer() -> int:
return tk_outer()
assert wf_outer() == sum(range(5))
Ketan (kumare3)
Yee
Robin Kahlow
06/16/2023, 7:16 PM