amoghrajesh commented on code in PR #54505:
URL: https://github.com/apache/airflow/pull/54505#discussion_r2556159487
##########
providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/operators/pod.py:
##########
@@ -81,13 +76,18 @@
PodNotFoundException,
PodPhase,
)
-from airflow.providers.cncf.kubernetes.version_compat import AIRFLOW_V_3_1_PLUS
+from airflow.providers.cncf.kubernetes.version_compat import (
+ AIRFLOW_V_3_1_PLUS,
+ AirflowSkipException,
+ TaskDeferred,
+)
from airflow.providers.common.compat.sdk import XCOM_RETURN_KEY
if AIRFLOW_V_3_1_PLUS:
from airflow.sdk import BaseOperator
else:
from airflow.models import BaseOperator
+from airflow.exceptions import AirflowException
Review Comment:
We try to keep providers importing from task sdk when there is a choice, I
would suggest if this can be made as `from
airflow.providers.cncf.kubernetes.version_compat import AirflowException` and
exporting it there.
--
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]
