uranusjr commented on code in PR #54693: URL: https://github.com/apache/airflow/pull/54693#discussion_r2314763686
########## devel-common/src/docs/shared/template-examples/taskflow-kwargs.rst: ########## @@ -17,15 +17,14 @@ .. code-block:: python - from airflow.models.taskinstance import TaskInstance - from airflow.models.dagrun import DagRun + from airflow.sdk.api.datamodels._generated import DagRun + from airflow.sdk.execution_time.task_runner import RuntimeTaskInstance Review Comment: We should probably expose some types (maybe in `airflow.sdk.types`) for users to import for type annotations like here. It is recommended too; since we expose Context as a public type, things in Context (and things in those things) should also be exposed somehow. -- 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]
