This is an automated email from the ASF dual-hosted git repository.
jscheffl pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new d96e4a19fa8 fix mypy error in provider (#60395)
d96e4a19fa8 is described below
commit d96e4a19fa89acfb959888e3803b0bbe68ffaeea
Author: Henry Chen <[email protected]>
AuthorDate: Mon Jan 12 04:41:43 2026 +0800
fix mypy error in provider (#60395)
---
.../src/airflow/providers/cncf/kubernetes/hooks/kubernetes.py | 7 ++++---
.../src/airflow/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 47ef247fac6..70ba19f1ff3 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 asyncio
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 @@ class AsyncKubernetesHook(KubernetesHook):
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 @@ class AsyncKubernetesHook(KubernetesHook):
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 865b5b67c15..e9e17f1b1bf 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 @@ from __future__ import annotations
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 @@ class GKEKubernetesAsyncHook(GoogleBaseAsyncHook,
AsyncKubernetesHook):
)
@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()