diff --git a/airflow/providers/amazon/aws/triggers/redshift_cluster.py b/airflow/providers/amazon/aws/triggers/redshift_cluster.py index 573aab4a5339e..ba67db01e641f 100644 --- a/airflow/providers/amazon/aws/triggers/redshift_cluster.py +++ b/airflow/providers/amazon/aws/triggers/redshift_cluster.py @@ -332,7 +332,7 @@ def serialize(self) -> tuple[str, dict[str, Any]]: def hook(self): return RedshiftHook(aws_conn_id=self.aws_conn_id) - async def run(self) -> AsyncIterator[TriggerEvent]: + async def run(self): async with self.hook.async_conn as client: attempt = 0 waiter = client.get_waiter("cluster_deleted") diff --git a/tests/providers/google/cloud/hooks/test_kubernetes_engine.py b/tests/providers/google/cloud/hooks/test_kubernetes_engine.py index 817a2e79e61ef..53963c3a329fe 100644 --- a/tests/providers/google/cloud/hooks/test_kubernetes_engine.py +++ b/tests/providers/google/cloud/hooks/test_kubernetes_engine.py @@ -17,7 +17,6 @@ # under the License. from __future__ import annotations -import sys from asyncio import Future from unittest import mock