From c7028f95f154e9b6aba66355329f28444a221830 Mon Sep 17 00:00:00 2001 From: Artur Shiriev Date: Wed, 7 Oct 2026 19:11:27 +0300 Subject: [PATCH] fix: annotate context managers with Generator for ty 0.0.85 --- modern_di_typer/main.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/modern_di_typer/main.py b/modern_di_typer/main.py index 91a0228..b7d05e7 100644 --- a/modern_di_typer/main.py +++ b/modern_di_typer/main.py @@ -1,3 +1,4 @@ +import collections.abc import contextlib import functools import inspect @@ -37,13 +38,13 @@ def fetch_di_container(ctx: typer.Context) -> Container: @contextlib.contextmanager -def _build_command_container(ctx: typer.Context) -> typing.Iterator[Container]: +def _build_command_container(ctx: typer.Context) -> collections.abc.Generator[Container]: with fetch_di_container(ctx).build_child_container(scope=Scope.REQUEST) as container: yield container @contextlib.contextmanager -def action_scope(ctx: typer.Context) -> typing.Iterator[Container]: +def action_scope(ctx: typer.Context) -> collections.abc.Generator[Container]: try: request_container: Container = ctx.meta[_COMMAND_CONTAINER_KEY] except KeyError: