From c8a40d97f1b8bcedeb9416560b9bc92f6448a96d Mon Sep 17 00:00:00 2001 From: ZipFile Date: Tue, 3 Jun 2025 16:40:39 +0000 Subject: [PATCH] fixup! Add support for async generator injections --- src/dependency_injector/_cwiring.pyx | 4 ++-- src/dependency_injector/wiring.py | 1 - 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/src/dependency_injector/_cwiring.pyx b/src/dependency_injector/_cwiring.pyx index a21ae165..3e2775c7 100644 --- a/src/dependency_injector/_cwiring.pyx +++ b/src/dependency_injector/_cwiring.pyx @@ -38,8 +38,8 @@ cdef class DependencyResolver: self.injections = injections self.closings = closings - async def _await_injection(self, p: KWPair, /) -> None: - self.to_inject[p.name] = await p.value + async def _await_injection(self, kw_pair: KWPair, /) -> None: + self.to_inject[kw_pair.name] = await kw_pair.value cdef object _await_injections(self, to_await: list): return gather(*map(self._await_injection, to_await)) diff --git a/src/dependency_injector/wiring.py b/src/dependency_injector/wiring.py index be17b8f8..aadf2cdc 100644 --- a/src/dependency_injector/wiring.py +++ b/src/dependency_injector/wiring.py @@ -1054,7 +1054,6 @@ def _get_async_patched(fn: F, patched: PatchedCallable) -> F: return cast(F, _patched) -# Async generators too... def _get_async_gen_patched(fn: F, patched: PatchedCallable) -> F: @functools.wraps(fn) async def _patched(*args: Any, **raw_kwargs: Any) -> AsyncIterator[Any]: