[GitHub] [airflow] ashb commented on a change in pull request #16289: Run mini scheduler in LocalTaskJob during task exit

[GitBox <gi...@apache.org>]

ashb commented on a change in pull request #16289:
URL: https://github.com/apache/airflow/pull/16289#discussion_r646377587



##########
File path: airflow/jobs/local_task_job.py
##########
@@ -149,6 +153,9 @@ def signal_handler(signum, frame):
     def handle_task_exit(self, return_code: int) -> None:
         """Handle case where self.task_runner exits by itself"""
         self.log.info("Task exited with return code %s", return_code)
+        if not self.task_instance.test_mode:
+            if conf.getboolean('scheduler', 'schedule_after_task_execution', fallback=True):
+                self._run_mini_scheduler_on_child_tasks()

Review comment:
       This should be at the end of this function, after all callbacks have been run.




-- 
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.

For queries about this service, please contact Infrastructure at:
users@infra.apache.org