Graph task

Conceptually, graph task let you treat a set of tasks as one task. They are similar to functions in programming since they can be reused and customized by changing their parameters.

Graph nodes can be nested, and they can also call themselves recursively.

Define a graph task with a decorator

You can define a graph task with the task.graph decorator. In the group definition you can also set default values for task properties.

from node_graph.decorator import task


@task()
def add(x, y):
    return x + y


@task()
def multiply(x, y):
    return x * y


@task.graph()
def add_multiply(x, y, z):
    add_out = add(x=x, y=y).result
    return multiply(x=z, y=add_out).result

Build the task group

ng = add_multiply.build(1, 2, 3)
ng

Use the graph task like a normal task

Create a task graph, add the group, set inputs, and run.

from node_graph import Graph

g1 = Graph(name="group_usage")
grp = g1.add_task(add_multiply, name="my_group")
g1