kaxil commented on a change in pull request #19950:
URL: https://github.com/apache/airflow/pull/19950#discussion_r765774322
##########
File path: airflow/jobs/local_task_job.py
##########
@@ -208,6 +208,13 @@ def heartbeat_callback(self, session=None):
)
raise AirflowException("PID of job runner does not match")
elif self.task_runner.return_code() is None and
hasattr(self.task_runner, 'process'):
+ if ti.state == State.SKIPPED:
+ # A DagRun timeout will cause tasks to be externally marked as
skipped.
+ dagrun = ti.get_dagrun()
Review comment:
```suggestion
dagrun = ti.get_dagrun(session=session)
```
--
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]
