From 3b128ee2e106daaf8756f2c80a18587a62797e95 Mon Sep 17 00:00:00 2001 From: sobolevn Date: Sat, 29 Jun 2024 13:17:41 +0300 Subject: [PATCH] Improve typing of `inject` --- rodi/__init__.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/rodi/__init__.py b/rodi/__init__.py index 7f27091..d043ff2 100644 --- a/rodi/__init__.py +++ b/rodi/__init__.py @@ -56,7 +56,10 @@ def __contains__(self, item) -> bool: AliasesTypeHint = Dict[str, Type] -def inject(globalsns=None, localns=None) -> Callable[..., Any]: +def inject( + globalsns: Optional[Dict[str, Any]] = None, + localns: Optional[Dict[str, Any]] = None, +) -> Callable[[T], T]: """ Marks a class or a function as injected. This method is only necessary if the class uses locals and the user uses Python >= 3.10, to bind the function's locals to the