amoghrajesh commented on code in PR #55767:
URL: https://github.com/apache/airflow/pull/55767#discussion_r2374979921
##########
task-sdk/src/airflow/sdk/execution_time/supervisor.py:
##########
@@ -1147,8 +1151,22 @@ def final_state(self):
return self._terminal_state or TaskInstanceState.SUCCESS
if self._exit_code != 0 and self._terminal_state == SERVER_TERMINATED:
return SERVER_TERMINATED
+
+ if self._exit_code != 0 and self._is_signal_retryable() and
self._should_retry:
+ return TaskInstanceState.UP_FOR_RETRY
Review Comment:
`final_state` translates exit code to the TI state in the end, just before
`update_task_state_if_needed` calls it and sends a client request, so it seemed
to be the most natural place to put this in.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
