fix retry/aioretry decorators type hints

Anvil · Dec 2, 2023 · 84c6251 · 84c6251
1 parent f6bf78f
commit 84c6251
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -10,6 +10,7 @@
 ### Changes
 
 ### Fixes
+* fix retry/aioretry decorators type hints
 
 ### Breaks
 

diff --git a/kaioretry/__init__.py b/kaioretry/__init__.py
@@ -5,7 +5,7 @@
 
 from typing import cast
 from collections.abc import Callable
-from mypy_extensions import DefaultNamedArg
+from mypy_extensions import DefaultNamedArg, DefaultArg
 
 from .types import Exceptions, NonNegative, Number, Jitter, \
     FuncParam, FuncRetVal, UpdateDelayFunc, JitterTuple, AioretryProtocol
@@ -56,7 +56,8 @@
 
 def _make_decorator(func: Callable[[Retry], Callable[FuncParam, FuncRetVal]]) \
     -> Callable[[
-        Exceptions, int,
+        DefaultArg(Exceptions, 'exceptions'),
+        DefaultArg(int, 'tries'),
         DefaultNamedArg(NonNegative, 'delay'),  # noqa: F821
         DefaultNamedArg(Number, 'backoff'),  # noqa: F821
         DefaultNamedArg(Jitter, 'jitter'),  # noqa: F821