diff --git a/babelizer/_utils.py b/babelizer/_utils.py index 7b18b67..af3d98e 100644 --- a/babelizer/_utils.py +++ b/babelizer/_utils.py @@ -72,7 +72,7 @@ def get_setup_py_version() -> str | None: @contextmanager -def save_files(files: Iterable[str]) -> Generator[dict[str, str], None, None]: +def save_files(files: Iterable[str]) -> Generator[dict[str, str]]: """Generate repository files through a context. Parameters @@ -96,7 +96,7 @@ def save_files(files: Iterable[str]) -> Generator[dict[str, str], None, None]: @contextmanager -def as_cwd(path: str) -> Generator[None, None, None]: +def as_cwd(path: str) -> Generator[None]: """Change directory context. Parameters diff --git a/babelizer/config.py b/babelizer/config.py index 1e466be..97b2589 100644 --- a/babelizer/config.py +++ b/babelizer/config.py @@ -84,7 +84,7 @@ def __init__( def __getitem__(self, key: str) -> dict[str, Any]: return self._meta[key] - def __iter__(self) -> Generator[str, None, None]: + def __iter__(self) -> Generator[str]: yield from self._meta def __len__(self) -> int: