def wf3(a: typing.List[int]): return map_task(lp)(a=a)
def w2(a: typing.List[int]) -> typing.List[str]: return map_task(complex_task, metadata=TaskMetadata(retries=2))(a=a)
def my_wf(a: typing.List[int]) -> int: x = map_task(t1, metadata=TaskMetadata(retries=1))(a=a) return t2(x=x)
def wf2(a: typing.List[int]): return map_task(wf1)(a=a)
def my_map_workflow(a: List[int]) -> str: mapped_out = map_task(my_pod_map_task, metadata=TaskMetadata(retries=1))(stringify=a) coalesced = coalesce(b=mapped_out) return coalesced
def my_wf(a: typing.List[str]) -> typing.List[str]: mappy = map_task(t1) map_node = create_node(mappy, a=a).with_overrides( requests=Resources(cpu="1", mem="100"), limits=Resources(cpu="2", mem="200")) return map_node.o0
def my_map_workflow(a: typing.List[int]) -> str: mapped_out = map_task(a_mappable_task, metadata=TaskMetadata(retries=1))(a=a) coalesced = coalesce(b=mapped_out) return coalesced
def test_map_task_metadata(): map_meta = TaskMetadata(retries=1) mapped_1 = map_task(t2, metadata=map_meta) assert mapped_1.metadata is map_meta mapped_2 = map_task(t2) assert mapped_2.metadata is t2.metadata
def my_wf(a: typing.List[str]) -> typing.List[str]: mappy = map_task(t1) map_node = mappy(a=a).with_overrides(limits=Resources(cpu="2", mem="200", ephemeral_storage="1Gi")) return map_node