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
