AIP-72: Logging task runner messages in supervisor (apache#45108)

PolideaInternal · Dec 20, 2024 · f10f552 · f10f552
1 parent 72ab6b4
commit f10f552
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/task_sdk/src/airflow/sdk/execution_time/supervisor.py b/task_sdk/src/airflow/sdk/execution_time/supervisor.py
@@ -709,6 +709,7 @@ def handle_requests(self, log: FilteringBoundLogger) -> Generator[None, bytes, N
             self._handle_request(msg, log)
 
     def _handle_request(self, msg: ToSupervisor, log: FilteringBoundLogger):
+        log.debug("Received message from task runner", msg=msg)
         resp = None
         if isinstance(msg, TaskState):
             self._terminal_state = msg.state