diff --git a/dagflow/dagflow.py b/dagflow/dagflow.py index 677ca4b..0832c35 100644 --- a/dagflow/dagflow.py +++ b/dagflow/dagflow.py @@ -260,7 +260,7 @@ where backid=${backid}$ elif ntype == 'subflow': await self.try_start_subflow(sor, inst, ne, node) elif ntype == 'start': - await self.node_transfer(sor, inst, ne, node, flow_def) + await self.node_transfer(sor, dsl, inst, ne, node, flow_def) await sor.U('node_execution', { 'id': ne.id, 'status': 'completed' @@ -278,7 +278,7 @@ where backid=${backid}$ }) break; elif status in ['done', 'cancelled', 'failed']: - await self.node_transfer(sor, inst, ne, node, flow_def) + await self.node_transfer(sor, dsl, inst, ne, node, flow_def) await sor.U('node_execution', { 'id': ne.id, 'status': 'completed'