|
26 | 26 | from tenacity import DoSleep |
27 | 27 | from tenacity import RetryCallState |
28 | 28 | from tenacity import RetryError |
29 | | -from tenacity import _utils |
30 | 29 | from tenacity import after_nothing |
31 | 30 | from tenacity import before_nothing |
32 | 31 | from tenacity import _utils |
@@ -57,16 +56,26 @@ def asyncio_sleep(duration: float) -> t.Awaitable[None]: |
57 | 56 | class AsyncRetrying(BaseRetrying): |
58 | 57 | def __init__( |
59 | 58 | self, |
60 | | - sleep: t.Callable[[t.Union[int, float]], t.Union[None, t.Awaitable[None]]] = asyncio_sleep, |
| 59 | + sleep: t.Callable[ |
| 60 | + [t.Union[int, float]], t.Union[None, t.Awaitable[None]] |
| 61 | + ] = asyncio_sleep, |
61 | 62 | stop: "StopBaseT" = tenacity.stop.stop_never, |
62 | 63 | wait: "WaitBaseT" = tenacity.wait.wait_none(), |
63 | 64 | retry: "t.Union[SyncRetryBaseT, RetryBaseT]" = tenacity.retry_if_exception_type(), |
64 | | - before: t.Callable[["RetryCallState"], t.Union[None, t.Awaitable[None]]] = before_nothing, |
65 | | - after: t.Callable[["RetryCallState"], t.Union[None, t.Awaitable[None]]] = after_nothing, |
66 | | - before_sleep: t.Optional[t.Callable[["RetryCallState"], t.Union[None, t.Awaitable[None]]]] = None, |
| 65 | + before: t.Callable[ |
| 66 | + ["RetryCallState"], t.Union[None, t.Awaitable[None]] |
| 67 | + ] = before_nothing, |
| 68 | + after: t.Callable[ |
| 69 | + ["RetryCallState"], t.Union[None, t.Awaitable[None]] |
| 70 | + ] = after_nothing, |
| 71 | + before_sleep: t.Optional[ |
| 72 | + t.Callable[["RetryCallState"], t.Union[None, t.Awaitable[None]]] |
| 73 | + ] = None, |
67 | 74 | reraise: bool = False, |
68 | 75 | retry_error_cls: t.Type["RetryError"] = RetryError, |
69 | | - retry_error_callback: t.Optional[t.Callable[["RetryCallState"], t.Union[t.Any, t.Awaitable[t.Any]]]] = None, |
| 76 | + retry_error_callback: t.Optional[ |
| 77 | + t.Callable[["RetryCallState"], t.Union[t.Any, t.Awaitable[t.Any]]] |
| 78 | + ] = None, |
70 | 79 | ) -> None: |
71 | 80 | super().__init__( |
72 | 81 | sleep=sleep, # type: ignore[arg-type] |
|
0 commit comments