Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 8 additions & 0 deletions flytekit/remote/remote.py
Original file line number Diff line number Diff line change
Expand Up @@ -1117,6 +1117,14 @@ def sync_node_execution(
"""
# For single task execution - the metadata spec node id is missing. In these cases, revert to regular node id
node_id = execution.metadata.spec_node_id
# This case supports single-task execution compiled workflows.
if node_id and node_id not in node_mapping and execution.id.node_id in node_mapping:
node_id = execution.id.node_id
remote_logger.debug(
f"Using node execution ID {node_id} instead of spec node id "
f"{execution.metadata.spec_node_id}, single-task execution likely."
)
# This case supports single-task execution compiled workflows with older versions of admin/propeller
if not node_id:
node_id = execution.id.node_id
remote_logger.debug(f"No metadata spec_node_id found, using {node_id}")
Expand Down