From 2c4d6fa0f4ac742450d800c340036edb24fadb45 Mon Sep 17 00:00:00 2001 From: Adam Dangoor Date: Fri, 24 Apr 2026 03:13:37 +0100 Subject: [PATCH] fix: use Generator instead of Iterator for contextmanager return types Pyright 1.1.409 deprecates annotating `@contextmanager` return types as `Iterator[Foo]`; it now requires `Generator[Foo]`. Co-Authored-By: Claude Opus 4.7 (1M context) --- src/vws_cli/commands.py | 4 ++-- src/vws_cli/query.py | 4 ++-- src/vws_cli/vumark.py | 4 ++-- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/src/vws_cli/commands.py b/src/vws_cli/commands.py index 81d5c32d..9f2ddd2a 100644 --- a/src/vws_cli/commands.py +++ b/src/vws_cli/commands.py @@ -4,7 +4,7 @@ import dataclasses import io import sys -from collections.abc import Iterator +from collections.abc import Generator from pathlib import Path import click @@ -40,7 +40,7 @@ @beartype @contextlib.contextmanager -def _handle_vws_exceptions() -> Iterator[None]: +def _handle_vws_exceptions() -> Generator[None]: """Show error messages and catch exceptions from ``VWS-Python``.""" error_message = "" diff --git a/src/vws_cli/query.py b/src/vws_cli/query.py index 3a517f49..e1648b77 100644 --- a/src/vws_cli/query.py +++ b/src/vws_cli/query.py @@ -4,7 +4,7 @@ import dataclasses import io import sys -from collections.abc import Iterator +from collections.abc import Generator from pathlib import Path import click @@ -35,7 +35,7 @@ @beartype @contextlib.contextmanager -def _handle_vwq_exceptions() -> Iterator[None]: +def _handle_vwq_exceptions() -> Generator[None]: """Show error messages and catch exceptions from ``VWS-Python``.""" try: yield diff --git a/src/vws_cli/vumark.py b/src/vws_cli/vumark.py index d4c6a92d..8fadb8f1 100644 --- a/src/vws_cli/vumark.py +++ b/src/vws_cli/vumark.py @@ -2,7 +2,7 @@ import contextlib import sys -from collections.abc import Iterator +from collections.abc import Generator from enum import StrEnum, unique from pathlib import Path @@ -51,7 +51,7 @@ class VuMarkFormatChoice(StrEnum): @beartype @contextlib.contextmanager -def _handle_vumark_exceptions() -> Iterator[None]: +def _handle_vumark_exceptions() -> Generator[None]: """Show error messages and catch exceptions from ``VWS-Python``.""" error_message = ""