From 6ae2da64e6d49a490b2fe5a43b957ae894d44546 Mon Sep 17 00:00:00 2001 From: Nikita Sobolev Date: Thu, 21 Jul 2022 18:51:13 +0300 Subject: [PATCH] Improve `redis.retry` types (#8354) --- stubs/redis/redis/retry.pyi | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/stubs/redis/redis/retry.pyi b/stubs/redis/redis/retry.pyi index beded9014..ec20224e8 100644 --- a/stubs/redis/redis/retry.pyi +++ b/stubs/redis/redis/retry.pyi @@ -1,6 +1,11 @@ -from collections.abc import Iterable +from collections.abc import Callable, Iterable +from typing import TypeVar + +from redis.backoff import AbstractBackoff + +_T = TypeVar("_T") class Retry: - def __init__(self, backoff, retries, supported_errors=...) -> None: ... - def update_supported_errors(self, specified_errors: Iterable[Exception]) -> None: ... - def call_with_retry(self, do, fail): ... + def __init__(self, backoff: AbstractBackoff, retries: int, supported_errors: tuple[type[Exception], ...] = ...) -> None: ... + def update_supported_errors(self, specified_errors: Iterable[type[Exception]]) -> None: ... + def call_with_retry(self, do: Callable[[], _T], fail: Callable[[Exception], object]) -> _T: ...