Paulo Benatto
01/29/2022, 11:21 AMfrom prefect import Flow, task
import prefect
import time
from prefect.executors import LocalDaskExecutor
logger = prefect.context.get("logger")
@task
def add_ten(x):
time.sleep(1)
return x + 10
@task
def save_number(y):
<http://logger.info|logger.info>(f"Saving number: {y}")
with Flow("reduce", executor=LocalDaskExecutor()) as flow:
mapped_result = add_ten.map([1, 2, 3, 4, 5, 6, 7, 8, 9, 9])
mapped_result_2 = add_ten.map(mapped_result)
reduced_result = save_number(mapped_result_2)
Anna Geller
01/29/2022, 11:52 AMwith Flow("reduce", executor=LocalExecutor()) as flow:
@task
def save_number(y):
<http://logger.info|logger.info>(f"Saving number: {y}")
with Flow("reduce", executor=LocalDaskExecutor()) as flow:
mapped_result = add_ten.map([1, 2, 3, 4, 5, 6, 7, 8, 9, 9])
mapped_result_2 = add_ten.map(mapped_result)
save_number.map(mapped_result_2)
Paulo Benatto
01/29/2022, 12:10 PM