diff --git a/providers/celery/src/airflow/providers/celery/executors/celery_kubernetes_executor.py b/providers/celery/src/airflow/providers/celery/executors/celery_kubernetes_executor.py index 7548401e295f0..04f4fa3b49c38 100644 --- a/providers/celery/src/airflow/providers/celery/executors/celery_kubernetes_executor.py +++ b/providers/celery/src/airflow/providers/celery/executors/celery_kubernetes_executor.py @@ -39,8 +39,8 @@ from airflow.callbacks.base_callback_sink import BaseCallbackSink from airflow.callbacks.callback_requests import CallbackRequest from airflow.executors.base_executor import EventBufferValueType - from airflow.models.taskinstance import ( - SimpleTaskInstance, # type: ignore[attr-defined] + from airflow.models.taskinstance import ( # type: ignore[attr-defined] + SimpleTaskInstance, TaskInstance, ) from airflow.models.taskinstancekey import TaskInstanceKey diff --git a/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/executors/local_kubernetes_executor.py b/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/executors/local_kubernetes_executor.py index de183ad2a4ae7..0cc7fa9dbc1f1 100644 --- a/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/executors/local_kubernetes_executor.py +++ b/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/executors/local_kubernetes_executor.py @@ -33,8 +33,8 @@ from airflow.callbacks.callback_requests import CallbackRequest from airflow.executors.base_executor import EventBufferValueType from airflow.executors.local_executor import LocalExecutor - from airflow.models.taskinstance import ( - SimpleTaskInstance, # type: ignore[attr-defined] + from airflow.models.taskinstance import ( # type: ignore[attr-defined] + SimpleTaskInstance, TaskInstance, TaskInstanceKey, )