Skip to content

Commit f0897be

Browse files
committed
fix: add failure node to non_system_nodes list
-e Signed-off-by: machichima <[email protected]>
1 parent 090f543 commit f0897be

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

flytekit/remote/entities.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -730,7 +730,9 @@ def promote_from_model(
730730
tasks: Optional[Dict[Identifier, FlyteTask]] = None,
731731
node_launch_plans: Optional[Dict[Identifier, launch_plan_models.LaunchPlanSpec]] = None,
732732
) -> FlyteWorkflow:
733-
base_model_non_system_nodes = cls.get_non_system_nodes(base_model.nodes)
733+
base_model_non_system_nodes = cls.get_non_system_nodes(base_model.nodes) + (
734+
[base_model.failure_node] if base_model.failure_node is not None else []
735+
)
734736

735737
node_map = {}
736738
converted_sub_workflows = {}

0 commit comments

Comments
 (0)