Run ruff format

jd · Mar 18, 2024 · 473c081 · 473c081
1 parent d0f11cb
commit 473c081
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 12 deletions.
diff --git a/tenacity/__init__.py b/tenacity/__init__.py
@@ -590,7 +590,8 @@ def __repr__(self) -> str:
 
 
 @t.overload
-def retry(func: WrappedFn) -> WrappedFn: ...
+def retry(func: WrappedFn) -> WrappedFn:
+    ...
 
 
 @t.overload
@@ -599,13 +600,22 @@ def retry(
     stop: "StopBaseT" = stop_never,
     wait: "WaitBaseT" = wait_none(),
     retry: "t.Union[RetryBaseT, tasyncio.retry.RetryBaseT]" = retry_if_exception_type(),
-    before: t.Callable[["RetryCallState"], t.Union[None, t.Awaitable[None]]] = before_nothing,
-    after: t.Callable[["RetryCallState"], t.Union[None, t.Awaitable[None]]] = after_nothing,
-    before_sleep: t.Optional[t.Callable[["RetryCallState"], t.Union[None, t.Awaitable[None]]]] = None,
+    before: t.Callable[
+        ["RetryCallState"], t.Union[None, t.Awaitable[None]]
+    ] = before_nothing,
+    after: t.Callable[
+        ["RetryCallState"], t.Union[None, t.Awaitable[None]]
+    ] = after_nothing,
+    before_sleep: t.Optional[
+        t.Callable[["RetryCallState"], t.Union[None, t.Awaitable[None]]]
+    ] = None,
     reraise: bool = False,
     retry_error_cls: t.Type["RetryError"] = RetryError,
-    retry_error_callback: t.Optional[t.Callable[["RetryCallState"], t.Union[t.Any, t.Awaitable[t.Any]]]] = None,
-) -> t.Callable[[WrappedFn], WrappedFn]: ...
+    retry_error_callback: t.Optional[
+        t.Callable[["RetryCallState"], t.Union[t.Any, t.Awaitable[t.Any]]]
+    ] = None,
+) -> t.Callable[[WrappedFn], WrappedFn]:
+    ...
 
 
 def retry(*dargs: t.Any, **dkw: t.Any) -> t.Any:

diff --git a/tenacity/asyncio/__init__.py b/tenacity/asyncio/__init__.py
@@ -26,7 +26,6 @@
 from tenacity import DoSleep
 from tenacity import RetryCallState
 from tenacity import RetryError
-from tenacity import _utils
 from tenacity import after_nothing
 from tenacity import before_nothing
 from tenacity import _utils
@@ -57,16 +56,26 @@ def asyncio_sleep(duration: float) -> t.Awaitable[None]:
 class AsyncRetrying(BaseRetrying):
     def __init__(
         self,
-        sleep: t.Callable[[t.Union[int, float]], t.Union[None, t.Awaitable[None]]] = asyncio_sleep,
+        sleep: t.Callable[
+            [t.Union[int, float]], t.Union[None, t.Awaitable[None]]
+        ] = asyncio_sleep,
         stop: "StopBaseT" = tenacity.stop.stop_never,
         wait: "WaitBaseT" = tenacity.wait.wait_none(),
         retry: "t.Union[SyncRetryBaseT, RetryBaseT]" = tenacity.retry_if_exception_type(),
-        before: t.Callable[["RetryCallState"], t.Union[None, t.Awaitable[None]]] = before_nothing,
-        after: t.Callable[["RetryCallState"], t.Union[None, t.Awaitable[None]]] = after_nothing,
-        before_sleep: t.Optional[t.Callable[["RetryCallState"], t.Union[None, t.Awaitable[None]]]] = None,
+        before: t.Callable[
+            ["RetryCallState"], t.Union[None, t.Awaitable[None]]
+        ] = before_nothing,
+        after: t.Callable[
+            ["RetryCallState"], t.Union[None, t.Awaitable[None]]
+        ] = after_nothing,
+        before_sleep: t.Optional[
+            t.Callable[["RetryCallState"], t.Union[None, t.Awaitable[None]]]
+        ] = None,
         reraise: bool = False,
         retry_error_cls: t.Type["RetryError"] = RetryError,
-        retry_error_callback: t.Optional[t.Callable[["RetryCallState"], t.Union[t.Any, t.Awaitable[t.Any]]]] = None,
+        retry_error_callback: t.Optional[
+            t.Callable[["RetryCallState"], t.Union[t.Any, t.Awaitable[t.Any]]]
+        ] = None,
     ) -> None:
         super().__init__(
             sleep=sleep,  # type: ignore[arg-type]