Skip to content

Commit a289bcd

Browse files
skip node in sync_node_execution if get node data returns missing error (#3344)
Signed-off-by: Yee Hing Tong <wild-endeavor@users.noreply.github.com>
1 parent 0663d8e commit a289bcd

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

flytekit/remote/remote.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2623,7 +2623,11 @@ def sync_node_execution(
26232623
raise ValueError(f"Missing node from mapping: {node_id}")
26242624

26252625
# Get the node execution data
2626-
node_execution_get_data_response = self.client.get_node_execution_data(execution.id)
2626+
try:
2627+
node_execution_get_data_response = self.client.get_node_execution_data(execution.id)
2628+
except FlyteEntityNotExistException:
2629+
logger.warning(f"Skipping node {execution.id.node_id} because node data not found")
2630+
return execution
26272631

26282632
# Calling a launch plan directly case
26292633
# If a node ran a launch plan directly (i.e. not through a dynamic task or anything) then

0 commit comments

Comments
 (0)