potiuk commented on code in PR #51153:
URL: https://github.com/apache/airflow/pull/51153#discussion_r2132347072
##########
providers/openlineage/src/airflow/providers/openlineage/utils/utils.py:
##########
@@ -68,9 +68,8 @@
from openlineage.client.event_v2 import Dataset as OpenLineageDataset
from openlineage.client.facet_v2 import RunFacet, processing_engine_run
- from airflow.models import TaskInstance
+ from airflow.models import DAG, MappedOperator, TaskInstance
from airflow.providers.common.compat.assets import Asset
- from airflow.sdk import DAG, MappedOperator
Review Comment:
BTW. This change likely comes from the `task sdk's.pyi` that is added in
this PR - so it proabably gives MyPy stronger signal of what is and what is not
available as attribute during TYPE_CHECKING.
--
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]
