schedula_dispatcher_quickstart_add_function_dataflow.py

1import schedula as sh
2
3# Create a Dispatcher
4dispatch = sh.Dispatcher(name='Quickstart Dispatcher')
5
6# Define a simple function to add two numbers
7def add(a, b):
8    return a + b
9
10# Add the function to the dispatcher
11# 'outputs' defines the name of the result in the data flow
12# 'inputs' defines the names of the arguments to be mapped
13dispatch.add_function(function=add, outputs=['c'], inputs=['a', 'b'])
14
15# Execute the dispatcher with initial inputs
16# 'a'=1 and 'b'=2
17solution = dispatch(inputs={'a': 1, 'b': 2})
18
19# Print the result stored in 'c'
20print(f"Result of a + b: {solution['c']}")