From d460ada5b38656b335a7ec865d367fe66489a86d Mon Sep 17 00:00:00 2001 From: henry3260 Date: Mon, 12 Jan 2026 02:38:29 +0800 Subject: [PATCH] fix mypy error in provider --- .../airflow/providers/cncf/kubernetes/hooks/kubernetes.py | 7 ++++--- .../providers/google/cloud/hooks/kubernetes_engine.py | 4 ++-- 2 files changed, 6 insertions(+), 5 deletions(-) diff --git a/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/hooks/kubernetes.py b/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/hooks/kubernetes.py index 47ef247fac6c2..70ba19f1ff34c 100644 --- a/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/hooks/kubernetes.py +++ b/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/hooks/kubernetes.py @@ -20,6 +20,7 @@ import contextlib import json import tempfile +from collections.abc import AsyncGenerator from functools import cached_property from time import sleep from typing import TYPE_CHECKING, Any, Protocol @@ -874,7 +875,7 @@ async def _get_field(self, field_name): return extras.get(prefixed_name) @contextlib.asynccontextmanager - async def get_conn(self) -> async_client.ApiClient: + async def get_conn(self) -> AsyncGenerator[async_client.ApiClient, None]: kube_client = None try: kube_client = await self._load_config() or async_client.ApiClient() @@ -951,8 +952,8 @@ async def read_logs( timestamps=True, since_seconds=since_seconds, ) - logs = logs.splitlines() - return logs + logs_list: list[str] = logs.splitlines() + return logs_list except HTTPError as e: raise KubernetesApiError from e diff --git a/providers/google/src/airflow/providers/google/cloud/hooks/kubernetes_engine.py b/providers/google/src/airflow/providers/google/cloud/hooks/kubernetes_engine.py index 865b5b67c1545..e9e17f1b1bf9d 100644 --- a/providers/google/src/airflow/providers/google/cloud/hooks/kubernetes_engine.py +++ b/providers/google/src/airflow/providers/google/cloud/hooks/kubernetes_engine.py @@ -22,7 +22,7 @@ import contextlib import json import time -from collections.abc import Sequence +from collections.abc import AsyncGenerator, Sequence from typing import TYPE_CHECKING, Any from google.api_core.exceptions import NotFound @@ -498,7 +498,7 @@ def __init__( ) @contextlib.asynccontextmanager - async def get_conn(self) -> async_client.ApiClient: + async def get_conn(self) -> AsyncGenerator[async_client.ApiClient, None]: kube_client = None try: kube_client = await self._load_config()