task-group-functions.py

from prefect import Flow, task, Parameter


@task
def foo():
    print("foo")


@task
def bar():
    print("bar")


@task
def plusone(x):
    return x + 1


def foobar():
    foo()
    bar()


def do_some_math(start_at):
    # Parameters can be passed to tasks
    result = plusone(start_at)

    # Logic cannot be performed unless in a task e.g.
    # if result > 2: ... not allowed

    return result


with Flow("example") as flow:
    start_at = Parameter("start_at", default=0)

    foobar()
    math_result = do_some_math(start_at)

    # Tasks can be called directly in the flow still
    plusone(math_result)


flow.run()