New RunLLM response generated:
>>>*# Question*
He...
# runllm-flyte-triala
ancient-wolf-19325
09/06/2024, 6:54 PMNew RunLLM response generated:
>>>*# Question*
Hey,
im currently working on a workflow template and my idea was to just swap out different functions as parameters where needed. This works as long as i just call the functions in my workflows but i cannot figure out a way to execute these functions as a map_task.
Copy code
from flytekit import dynamic, task, workflow, map_task, PythonFunctionTask
@task
def t1(foo: list[int]) -> None:
for i in foo:
print(i)
@task
def t2(bar: list[int]) -> None:
for i in bar:
print(i)
@dynamic
def d1(task_function_to_map: PythonFunctionTask, l: dict[str, list[int]]) -> None:
# Do common stuff
map_task(task_function_to_map)(**l)
# Do more common stuff
@workflow
def main_wf() -> None:
d1(
task_function_to_map=t1,
l={"foo": [1, 2, 3, 4, 5, 6, 7, 8, 9]}
)
d1(
task_function_to_map=t2,
l={"bar": [1, 2, 3, 4, 5, 6, 7, 8, 9]}
)map_taskmap_taskArrayNodeArrayNode