From ff29d849953c0fa3d62415a84c8eb290516529f4 Mon Sep 17 00:00:00 2001 From: Roman Mogylatov Date: Fri, 13 Nov 2020 17:07:40 -0500 Subject: [PATCH] Add generic typing for @inject --- src/dependency_injector/wiring.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/dependency_injector/wiring.py b/src/dependency_injector/wiring.py index 9f12985f..953f3a1f 100644 --- a/src/dependency_injector/wiring.py +++ b/src/dependency_injector/wiring.py @@ -28,6 +28,7 @@ __all__ = ( ) T = TypeVar('T') +F = TypeVar('F', bound=Callable[..., Any]) Container = Any @@ -203,7 +204,7 @@ def unwire( _unpatch(member, method_name, method) -def inject(fn: Callable[..., Any]) -> Callable[..., Any]: +def inject(fn: F) -> F: """Decorate callable with injecting decorator.""" reference_injections, reference_closing = _fetch_reference_injections(fn) return _get_patched(fn, reference_injections, reference_closing)