elegant-petabyte-32634
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))
freezing-airport-6809
thankful-minister-83577
thankful-minister-83577
elegant-petabyte-32634
06/16/2023, 7:16 PMelegant-petabyte-32634
06/16/2023, 7:17 PMelegant-petabyte-32634
06/16/2023, 7:18 PM