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 = ""