ashb commented on code in PR #56955:
URL: https://github.com/apache/airflow/pull/56955#discussion_r2461986661
##########
task-sdk/src/airflow/sdk/execution_time/task_runner.py:
##########
@@ -550,6 +552,16 @@ def get_dagrun_state(dag_id: str, run_id: str) -> str:
return response.state
+ @staticmethod
+ def get_dag_state(dag_id: str) -> DagStateResult:
+ """Return the state of the Dag run with the given Run ID."""
+ response = SUPERVISOR_COMMS.send(msg=GetDagState(dag_id=dag_id))
+
+ if TYPE_CHECKING:
+ assert isinstance(response, DagStateResult)
+
+ return response
+
Review Comment:
What is going to call this?
--
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]
