diff --git a/exception_escaping/proxy_module.py b/exception_escaping/proxy_module.py index 0dca7a6..fd2210d 100644 --- a/exception_escaping/proxy_module.py +++ b/exception_escaping/proxy_module.py @@ -5,7 +5,7 @@ class ProxyModule(sys.modules[__name__].__class__): # type: ignore[misc] - def __call__(self, *args: Callable[..., Any], default_return: Any = None) -> Union[Callable[..., Any], Callable[Callable[..., Any], Callable[..., Any]]]: + def __call__(self, *args: Callable[..., Any], default_return: Any = None) -> Union[Callable[..., Any], Callable[[Callable[..., Any]], Callable[..., Any]]]: def wrapper_of_wrappers(function: Callable[..., Any]) -> Callable[..., Any]: @wraps(function) def wrapper(*args: Any, **kwargs: Any) -> Any: