From f7186f81a394869d924608bf41cafe34fbf77091 Mon Sep 17 00:00:00 2001 From: Pavan Kumar Date: Sat, 12 Jul 2025 21:27:59 +0100 Subject: [PATCH] Cleanup mypy ignores in celery provider --- .../celery/src/airflow/providers/celery/cli/celery_command.py | 2 +- .../providers/celery/executors/celery_executor_utils.py | 2 +- .../providers/celery/executors/celery_kubernetes_executor.py | 4 ++-- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/providers/celery/src/airflow/providers/celery/cli/celery_command.py b/providers/celery/src/airflow/providers/celery/cli/celery_command.py index 997cbb4c8a56d..c584cba1b76d3 100644 --- a/providers/celery/src/airflow/providers/celery/cli/celery_command.py +++ b/providers/celery/src/airflow/providers/celery/cli/celery_command.py @@ -27,7 +27,7 @@ import psutil import sqlalchemy.exc -from celery import maybe_patch_concurrency # type: ignore[attr-defined] +from celery import maybe_patch_concurrency from celery.app.defaults import DEFAULT_TASK_LOG_FMT from celery.signals import after_setup_logger from lockfile.pidlockfile import read_pid_from_pidfile, remove_existing_pidfile diff --git a/providers/celery/src/airflow/providers/celery/executors/celery_executor_utils.py b/providers/celery/src/airflow/providers/celery/executors/celery_executor_utils.py index f3120e84e7ee1..e85a0f96239b8 100644 --- a/providers/celery/src/airflow/providers/celery/executors/celery_executor_utils.py +++ b/providers/celery/src/airflow/providers/celery/executors/celery_executor_utils.py @@ -286,7 +286,7 @@ def send_task_to_executor( # The type is right for the version, but the type cannot be defined correctly for Airflow 2 and 3 # concurrently; - return key, args, result # type: ignore[return-value] + return key, args, result def fetch_celery_task_state(async_result: AsyncResult) -> tuple[str, str | ExceptionWithTraceback, Any]: 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 04f4fa3b49c38..78025db44d038 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 @@ -108,9 +108,9 @@ def _task_event_logs(self, value): def queued_tasks(self) -> dict[TaskInstanceKey, Any]: """Return queued tasks from celery and kubernetes executor.""" queued_tasks = self.celery_executor.queued_tasks.copy() - queued_tasks.update(self.kubernetes_executor.queued_tasks) # type: ignore[arg-type] + queued_tasks.update(self.kubernetes_executor.queued_tasks) - return queued_tasks # type: ignore[return-value] + return queued_tasks @queued_tasks.setter def queued_tasks(self, value) -> None: