Michael Smith
03/28/2022, 7:44 PMKevin Kho
Zanie
future = my_task(arg, arg)
state = future.wait(5)
if not state:
# Do something based on this timeout
...
Wait N sconds for the task to complete
>>> @flow
>>> def my_flow():
>>> future = my_task()
>>> final_state = future.wait(0.1)
>>> if final_state:
>>> ... # Task done
>>> else:
>>> ... # Task not done yet
Michael Smith
03/29/2022, 7:24 AMfrom prefect import flow, task
from time import sleep
@task
def do_something():
print("Do something")
sleep(20)
@flow(timeout_seconds=10)
def flow2():
do_something()
@flow(timeout_seconds=60)
def timeout_task():
do_something()
flow2()
timeout_task()
from prefect import flow, task
from time import sleep
@flow(timeout_seconds=10)
def combined():
@task
def combined_inner():
print("Do something")
sleep(20)
combined_inner()
combined()
Kevin Kho