This is an automated email from the ASF dual-hosted git repository.
uranusjr 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 7dd0d84d14 Add typing to concrete taskflow examples (#33417)
7dd0d84d14 is described below
commit 7dd0d84d145114d3de4585769b19c1e737485830
Author: Jens Scheffler <[email protected]>
AuthorDate: Wed Aug 16 08:21:27 2023 +0200
Add typing to concrete taskflow examples (#33417)
---
docs/shared/template-examples/taskflow-kwargs.rst | 8 ++++++--
docs/shared/template-examples/taskflow.rst | 6 +++++-
2 files changed, 11 insertions(+), 3 deletions(-)
diff --git a/docs/shared/template-examples/taskflow-kwargs.rst
b/docs/shared/template-examples/taskflow-kwargs.rst
index 7a4c27c566..ed92b991ff 100644
--- a/docs/shared/template-examples/taskflow-kwargs.rst
+++ b/docs/shared/template-examples/taskflow-kwargs.rst
@@ -17,11 +17,15 @@
.. code-block:: python
+ from airflow.models.taskinstance import TaskInstance
+ from airflow.models.dagrun import DagRun
+
+
@task
def print_ti_info(**kwargs):
- ti = kwargs["task_instance"]
+ 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
- dr = kwargs["dag_run"]
+ dr: DagRun = kwargs["dag_run"]
print(f"DAG Run queued at: {dr.queued_at}") # 2023-08-10
00:00:01+02:20
diff --git a/docs/shared/template-examples/taskflow.rst
b/docs/shared/template-examples/taskflow.rst
index a9d46e1c63..5ea6955451 100644
--- a/docs/shared/template-examples/taskflow.rst
+++ b/docs/shared/template-examples/taskflow.rst
@@ -17,8 +17,12 @@
.. code-block:: python
+ from airflow.models.taskinstance import TaskInstance
+ from airflow.models.dagrun import DagRun
+
+
@task
- def print_ti_info(task_instance=None, dag_run=None):
+ def print_ti_info(task_instance: TaskInstance | None = None, dag_run:
DagRun | None = None):
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
