diff --git a/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/executors/kubernetes_executor.py b/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/executors/kubernetes_executor.py index 37908c176e616..2f8ff5699e3da 100644 --- a/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/executors/kubernetes_executor.py +++ b/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/executors/kubernetes_executor.py @@ -307,7 +307,7 @@ def _process_workloads(self, workloads: Sequence[workloads.All]) -> None: executor_config = w.ti.executor_config or {} del self.queued_tasks[key] - self.execute_async(key=key, command=command, queue=queue, executor_config=executor_config) # type: ignore[arg-type] + self.execute_async(key=key, command=command, queue=queue, executor_config=executor_config) self.running.add(key) def sync(self) -> None: diff --git a/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/operators/job.py b/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/operators/job.py index 8e7ef694b9cbe..7c19421594aa2 100644 --- a/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/operators/job.py +++ b/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/operators/job.py @@ -209,10 +209,10 @@ def execute(self, context: Context): def execute_deferrable(self): self.defer( trigger=KubernetesJobTrigger( - job_name=self.job.metadata.name, # type: ignore[union-attr] - job_namespace=self.job.metadata.namespace, # type: ignore[union-attr] - pod_name=self.pod.metadata.name, # type: ignore[union-attr] - pod_namespace=self.pod.metadata.namespace, # type: ignore[union-attr] + job_name=self.job.metadata.name, + job_namespace=self.job.metadata.namespace, + pod_name=self.pod.metadata.name, + pod_namespace=self.pod.metadata.namespace, base_container_name=self.base_container_name, kubernetes_conn_id=self.kubernetes_conn_id, cluster_context=self.cluster_context, diff --git a/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/operators/pod.py b/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/operators/pod.py index 994c54822a7be..0c310395b9821 100644 --- a/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/operators/pod.py +++ b/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/operators/pod.py @@ -68,7 +68,7 @@ ) from airflow.providers.cncf.kubernetes.pod_generator import PodGenerator from airflow.providers.cncf.kubernetes.triggers.pod import KubernetesPodTrigger -from airflow.providers.cncf.kubernetes.utils import xcom_sidecar # type: ignore[attr-defined] +from airflow.providers.cncf.kubernetes.utils import xcom_sidecar from airflow.providers.cncf.kubernetes.utils.pod_manager import ( EMPTY_XCOM_RESULT, OnFinishAction, diff --git a/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/triggers/job.py b/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/triggers/job.py index a6b03e9fdea53..359c30547881c 100644 --- a/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/triggers/job.py +++ b/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/triggers/job.py @@ -100,7 +100,7 @@ def serialize(self) -> tuple[str, dict[str, Any]]: }, ) - async def run(self) -> AsyncIterator[TriggerEvent]: # type: ignore[override] + async def run(self) -> AsyncIterator[TriggerEvent]: """Get current job status and yield a TriggerEvent.""" if self.get_logs or self.do_xcom_push: pod = await self.hook.get_pod(name=self.pod_name, namespace=self.pod_namespace) diff --git a/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/triggers/pod.py b/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/triggers/pod.py index 719e1f6f34441..2646854c5d04d 100644 --- a/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/triggers/pod.py +++ b/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/triggers/pod.py @@ -141,7 +141,7 @@ def serialize(self) -> tuple[str, dict[str, Any]]: }, ) - async def run(self) -> AsyncIterator[TriggerEvent]: # type: ignore[override] + async def run(self) -> AsyncIterator[TriggerEvent]: """Get current pod status and yield a TriggerEvent.""" self.log.info("Checking pod %r in namespace %r.", self.pod_name, self.pod_namespace) try: diff --git a/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/version_compat.py b/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/version_compat.py index fd909dbbcfbed..2a974225f0f96 100644 --- a/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/version_compat.py +++ b/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/version_compat.py @@ -40,8 +40,8 @@ def get_base_airflow_version_tuple() -> tuple[int, int, int]: from airflow.sdk import BaseHook, BaseOperator else: from airflow.hooks.base import BaseHook # type: ignore[attr-defined,no-redef] - from airflow.models import BaseOperator # type: ignore[no-redef] - from airflow.utils.xcom import XCOM_RETURN_KEY # type: ignore[no-redef] + from airflow.models import BaseOperator + from airflow.utils.xcom import XCOM_RETURN_KEY if AIRFLOW_V_3_0_PLUS: from airflow.sdk import BaseSensorOperator