CJ
08/08/2022, 6:54 PMclaire
08/08/2022, 7:03 PM@op
def upstream_op():
return 5
@op
def downstream_op(my_input):
return my_input
@graph
def my_graph(my_input):
downstream_op(my_input)
@job
def my_job():
my_graph(upstream_op())
claire
08/08/2022, 7:04 PMCJ
08/08/2022, 7:08 PMCJ
08/08/2022, 7:11 PMclaire
08/08/2022, 7:16 PMrun_config
. For example:
@op
def downstream_op(my_input):
return my_input
@graph
def my_graph(my_input):
downstream_op(my_input)
my_graph.execute_in_process(
run_config={"inputs": {"my_input": {"value": 8}}}
)
CJ
08/08/2022, 7:20 PMclaire
08/08/2022, 8:01 PM@op
def downstream_op(my_input):
return my_input
@graph
def my_graph(my_input):
downstream_op(my_input)
@op(out=DynamicOut())
def dynamic_outs():
for i in range(5):
yield DynamicOutput(i, mapping_key=str(i))
@job
def my_job():
dynamic_outs().map(my_graph)
Dynamic outputs will allow you to return an arbitrary number of outputs from an upstream op, and then you can map these outputs to be inputs to your graph.CJ
08/08/2022, 8:02 PM