From 78775aadb46cb6d9f3af0085d9ea5edf0fc5ea85 Mon Sep 17 00:00:00 2001 From: sobolevn Date: Wed, 28 Jun 2023 18:34:03 +0300 Subject: [PATCH 1/2] Use propert return types for `__await__` methods in `redis.asyncio.client` --- stubs/redis/redis/asyncio/client.pyi | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/stubs/redis/redis/asyncio/client.pyi b/stubs/redis/redis/asyncio/client.pyi index 451b7b7ee510..26efc2b58ec3 100644 --- a/stubs/redis/redis/asyncio/client.pyi +++ b/stubs/redis/redis/asyncio/client.pyi @@ -1,5 +1,5 @@ from _typeshed import Incomplete, Unused -from collections.abc import AsyncIterator, Awaitable, Callable, Iterable, Mapping, MutableMapping, Sequence +from collections.abc import AsyncIterator, Awaitable, Callable, Iterable, Mapping, MutableMapping, Sequence, Generator from datetime import datetime, timedelta from types import TracebackType from typing import Any, ClassVar, Generic, NoReturn, Protocol, overload @@ -67,7 +67,7 @@ class Redis(AbstractRedis, RedisModuleCommands, AsyncCoreCommands[_StrType], Asy redis_connect_func: ConnectCallbackT | None = None, credential_provider: CredentialProvider | None = None, ) -> None: ... - def __await__(self): ... + def __await__(self) -> Generator[Any, None, Self]: ... async def initialize(self) -> Self: ... def set_response_callback(self, command: str, callback: ResponseCallbackT): ... def load_external_module(self, funcname, func) -> None: ... @@ -200,7 +200,7 @@ class Pipeline(Redis[_StrType], Generic[_StrType]): async def __aexit__( self, exc_type: type[BaseException] | None, exc_value: BaseException | None, traceback: TracebackType | None ) -> None: ... - def __await__(self): ... + def __await__(self) -> Generator[Any, None, Self]: ... def __len__(self) -> int: ... def __bool__(self) -> bool: ... async def reset(self) -> None: ... # type: ignore[override] From 468493e3d0956bb80e9a6ee804cd1d1bf92202b8 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Wed, 28 Jun 2023 15:35:50 +0000 Subject: [PATCH 2/2] [pre-commit.ci] auto fixes from pre-commit.com hooks --- stubs/redis/redis/asyncio/client.pyi | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/stubs/redis/redis/asyncio/client.pyi b/stubs/redis/redis/asyncio/client.pyi index 26efc2b58ec3..2ec11d7b5120 100644 --- a/stubs/redis/redis/asyncio/client.pyi +++ b/stubs/redis/redis/asyncio/client.pyi @@ -1,5 +1,5 @@ from _typeshed import Incomplete, Unused -from collections.abc import AsyncIterator, Awaitable, Callable, Iterable, Mapping, MutableMapping, Sequence, Generator +from collections.abc import AsyncIterator, Awaitable, Callable, Generator, Iterable, Mapping, MutableMapping, Sequence from datetime import datetime, timedelta from types import TracebackType from typing import Any, ClassVar, Generic, NoReturn, Protocol, overload