This is an automated email from the ASF dual-hosted git repository.
potiuk pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new fc40cfb1dbe Fix TaskFlow context docs example (#66698)
fc40cfb1dbe is described below
commit fc40cfb1dbeebd71fedb12a07bc2c932901a9b4c
Author: Kyu Park <[email protected]>
AuthorDate: Tue May 12 01:41:40 2026 +0200
Fix TaskFlow context docs example (#66698)
Replace invalid TaskFlow context attributes with fields available on the
runtime TaskInstance and DagRun protocol.
---
.../src/docs/shared/template-examples/taskflow-kwargs.rst | 10 +++++-----
devel-common/src/docs/shared/template-examples/taskflow.rst | 10 +++++-----
2 files changed, 10 insertions(+), 10 deletions(-)
diff --git a/devel-common/src/docs/shared/template-examples/taskflow-kwargs.rst
b/devel-common/src/docs/shared/template-examples/taskflow-kwargs.rst
index c8dd9e23ee5..92ed80175d5 100644
--- a/devel-common/src/docs/shared/template-examples/taskflow-kwargs.rst
+++ b/devel-common/src/docs/shared/template-examples/taskflow-kwargs.rst
@@ -17,15 +17,15 @@
.. code-block:: python
- from airflow.models.taskinstance import TaskInstance
- from airflow.models.dagrun import DagRun
+ from airflow.sdk import TaskInstance
+ from airflow.sdk.types import DagRunProtocol
@task
def print_ti_info(**kwargs):
ti: TaskInstance = kwargs["task_instance"]
print(f"Run ID: {ti.run_id}") # Run ID:
scheduled__2023-08-09T00:00:00+00:00
- print(f"Duration: {ti.duration}") # Duration: 0.972019
+ print(f"Task start date: {ti.start_date}") # 2023-08-10 00:00:01+00:00
- dr: DagRun = kwargs["dag_run"]
- print(f"Dag Run queued at: {dr.queued_at}") # 2023-08-10
00:00:01+02:20
+ dr: DagRunProtocol = kwargs["dag_run"]
+ print(f"Dag Run logical date: {dr.logical_date}") # 2023-08-09
00:00:00+00:00
diff --git a/devel-common/src/docs/shared/template-examples/taskflow.rst
b/devel-common/src/docs/shared/template-examples/taskflow.rst
index 2caecaa4b11..0a460a728e2 100644
--- a/devel-common/src/docs/shared/template-examples/taskflow.rst
+++ b/devel-common/src/docs/shared/template-examples/taskflow.rst
@@ -17,12 +17,12 @@
.. code-block:: python
- from airflow.models.taskinstance import TaskInstance
- from airflow.models.dagrun import DagRun
+ from airflow.sdk import TaskInstance
+ from airflow.sdk.types import DagRunProtocol
@task
- def print_ti_info(task_instance: TaskInstance, dag_run: DagRun):
+ def print_ti_info(task_instance: TaskInstance, dag_run: DagRunProtocol):
print(f"Run ID: {task_instance.run_id}") # Run ID:
scheduled__2023-08-09T00:00:00+00:00
- print(f"Duration: {task_instance.duration}") # Duration: 0.972019
- print(f"Dag Run queued at: {dag_run.queued_at}") # 2023-08-10
00:00:01+02:20
+ print(f"Task start date: {task_instance.start_date}") # 2023-08-10
00:00:01+00:00
+ print(f"Dag Run logical date: {dag_run.logical_date}") # 2023-08-09
00:00:00+00:00
