typing: fix aretry typing to match docstring (#128)

quora · Oct 10, 2023 · cc3567e · cc3567e
1 parent cf6ebcd
commit cc3567e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/asynq/tools.pyi b/asynq/tools.pyi
@@ -11,7 +11,9 @@ from typing import (
     List,
     Optional,
     Tuple,
+    Type,
     TypeVar,
+    Union,
     overload,
 )
 
@@ -62,7 +64,9 @@ def alru_cache(
 ) -> _AsyncWrapper: ...
 def alazy_constant(ttl: int = ...) -> Callable[..., Any]: ...
 def aretry(
-    exception_cls: type, max_tries: int = ..., sleep: float = ...
+    exception_cls: Union[Type[BaseException], Tuple[Type[BaseException], ...]],
+    max_tries: int = ...,
+    sleep: float = ...,
 ) -> Callable[..., Any]: ...
 @asynq()
 def call_with_context(