Skip to content

Commit 0c630fa

Browse files
handle KubernetesDeleteJobOperator import (#39036)
1 parent 1ded297 commit 0c630fa

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

airflow/providers/google/cloud/operators/kubernetes_engine.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@
3535

3636
from airflow.configuration import conf
3737
from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
38-
from airflow.providers.cncf.kubernetes.operators.job import KubernetesDeleteJobOperator, KubernetesJobOperator
38+
from airflow.providers.cncf.kubernetes.operators.job import KubernetesJobOperator
3939
from airflow.providers.cncf.kubernetes.operators.pod import KubernetesPodOperator
4040
from airflow.providers.cncf.kubernetes.operators.resource import (
4141
KubernetesCreateResourceOperator,
@@ -65,6 +65,16 @@
6565
from airflow.providers_manager import ProvidersManager
6666
from airflow.utils.timezone import utcnow
6767

68+
try:
69+
from airflow.providers.cncf.kubernetes.operators.job import KubernetesDeleteJobOperator
70+
except ImportError:
71+
from airflow.exceptions import AirflowOptionalProviderFeatureException
72+
73+
raise AirflowOptionalProviderFeatureException(
74+
"Failed to import KubernetesDeleteJobOperator. This operator is only available in cncf-kubernetes "
75+
"provider version >=8.1.0"
76+
)
77+
6878
if TYPE_CHECKING:
6979
from kubernetes.client.models import V1Job, V1Pod
7080

0 commit comments

Comments
 (0)