chore: pass DataFlowGraph instead of Function as arg
#9656
Merged
DataFlowGraph instead of Function as arg
#9656