Add type cast for @inject

This commit is contained in:
Roman Mogylatov 2020-11-13 17:24:35 -05:00
parent ff29d84995
commit fb47d5db0d

View File

@ -207,7 +207,8 @@ def unwire(
def inject(fn: F) -> F: def inject(fn: F) -> F:
"""Decorate callable with injecting decorator.""" """Decorate callable with injecting decorator."""
reference_injections, reference_closing = _fetch_reference_injections(fn) reference_injections, reference_closing = _fetch_reference_injections(fn)
return _get_patched(fn, reference_injections, reference_closing) patched = _get_patched(fn, reference_injections, reference_closing)
return cast(F, patched)
def _patch_fn( def _patch_fn(