Add support for async sleep functions in tenacity.retry annotation. F…

…ixes #399 (#400)
jd · Aug 14, 2023 · 41ed242 · 41ed242
1 parent aa6f8f0
commit 41ed242
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tenacity/__init__.py b/tenacity/__init__.py
@@ -499,7 +499,7 @@ def retry(func: WrappedFn) -> WrappedFn:
 
 @t.overload
 def retry(
-    sleep: t.Callable[[t.Union[int, float]], None] = sleep,
+    sleep: t.Callable[[t.Union[int, float]], t.Optional[t.Awaitable[None]]] = sleep,
     stop: "StopBaseT" = stop_never,
     wait: "WaitBaseT" = wait_none(),
     retry: "RetryBaseT" = retry_if_exception_type(),