From f78bb88940818261c3b525ba18b06284b807286f Mon Sep 17 00:00:00 2001 From: deependujha Date: Fri, 25 Sep 2026 12:41:21 +0530 Subject: [PATCH] fix: correct _suppress_os_stderr generator type annotation --- src/litmodels/io/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/litmodels/io/utils.py b/src/litmodels/io/utils.py index 32f490c..ebaf0ca 100644 --- a/src/litmodels/io/utils.py +++ b/src/litmodels/io/utils.py @@ -1,6 +1,6 @@ import os import pickle -from collections.abc import Iterator +from collections.abc import Generator from contextlib import contextmanager from pathlib import Path from typing import Any, Union @@ -10,7 +10,7 @@ @contextmanager -def _suppress_os_stderr() -> Iterator[None]: +def _suppress_os_stderr() -> Generator[None, None, None]: devnull_fd = os.open(os.devnull, os.O_WRONLY) old_stderr_fd = os.dup(2) os.dup2(devnull_fd, 2) # redirect stderr (fd 2) to /dev/null