Subham-KRLX opened a new pull request, #61568:
URL: https://github.com/apache/airflow/pull/61568
This PR exposes
TaskInstance
in the public API of the Task SDK.
The
TaskInstance
class (which is a runtime representation, distinct from the ORM model) was
documented in the public API reference but was not actually exposed in the
airflow.sdk module.
This change:
Adds a public alias TaskInstance = RuntimeTaskInstanceProtocol in
airflow/sdk/types.py
.
Exports
TaskInstance
in
airflow/sdk/
init
.py
(in __all__, TYPE_CHECKING, and lazy imports).
This allows users to correctly import and type hint
TaskInstance
from the SDK:
python
from airflow.sdk import TaskInstance
Related Issue
Closes: #60478
Tests
Validated locally that from airflow.sdk import TaskInstance works.
Checked
type(TaskInstance)
confirms it is
RuntimeTaskInstanceProtocol
.
Was generative AI tooling used to co-author this PR?
Yes (please specify the tool below)
Good
Bad
--
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]
