enforce callable type in __call__ return value

Anvil · Nov 24, 2023 · 0bb81c6 · 0bb81c6
1 parent bd6fed0
commit 0bb81c6
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/kaioretry/decorator.py b/kaioretry/decorator.py
@@ -295,8 +295,7 @@ def is_func_async(cls, func: Function) -> bool:
     def __call__(self, func: Callable[FuncParam, FuncRetVal]) \
             -> Callable[FuncParam, FuncRetVal]:
         if self.is_func_async(func):
-            return self.aioretry(
-                cast(Callable[FuncParam, Awaitable[Any]], func))
+            return cast(Callable[FuncParam, FuncRetVal], self.aioretry(func))
         return self.retry(func)
 
     def __str__(self) -> str: