Use airflow.models.dag.DAG in k8s tests (apache#34616)

asapekia · Sep 26, 2023 · 11cba87 · 11cba87
1 parent 4df5b57
commit 11cba87
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/kubernetes_tests/test_kubernetes_pod_operator.py b/kubernetes_tests/test_kubernetes_pod_operator.py
@@ -34,7 +34,10 @@
 from kubernetes.client.rest import ApiException
 
 from airflow.exceptions import AirflowException, AirflowSkipException
-from airflow.models import DAG, Connection, DagRun, TaskInstance
+from airflow.models.connection import Connection
+from airflow.models.dag import DAG
+from airflow.models.dagrun import DagRun
+from airflow.models.taskinstance import TaskInstance
 from airflow.providers.cncf.kubernetes.hooks.kubernetes import KubernetesHook
 from airflow.providers.cncf.kubernetes.operators.pod import KubernetesPodOperator
 from airflow.providers.cncf.kubernetes.utils.pod_manager import OnFinishAction, PodManager