Skip to content

Commit d87a800

Browse files
committed
refactor: pass failure_node to get_non_system_nodes function
-e Signed-off-by: machichima <[email protected]>
1 parent e2cae26 commit d87a800

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

flytekit/remote/entities.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -730,9 +730,7 @@ 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) + (
734-
[base_model.failure_node] if base_model.failure_node is not None else []
735-
)
733+
base_model_non_system_nodes = cls.get_non_system_nodes(base_model.nodes + [base_model.failure_node])
736734

737735
node_map = {}
738736
converted_sub_workflows = {}

0 commit comments

Comments
 (0)