Timeouts and Retries

from beam import function


@function(timeout=600)
def handler():
    return sum(range(1_000_000))


if __name__ == "__main__":
    print(handler.remote())

from beam import TaskPolicy, task_queue


@task_queue(timeout=1800, task_policy=TaskPolicy(ttl=900))
def process(**inputs):
    return {"received": inputs}

from beam import task_queue


@task_queue(retries=2, retry_for=[ConnectionError])
def process(**inputs):
    # Raise ConnectionError for a transient dependency failure.
    return {"received": inputs}