From 513861bf9ad0254f75671f05f08631c397e94a9b Mon Sep 17 00:00:00 2001 From: Samuel Monson Date: Tue, 18 Nov 2025 17:29:02 -0500 Subject: [PATCH] Unmask StopIteration Signed-off-by: Samuel Monson --- src/guidellm/data/loaders.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/guidellm/data/loaders.py b/src/guidellm/data/loaders.py index fbeaf0583..e6393707e 100644 --- a/src/guidellm/data/loaders.py +++ b/src/guidellm/data/loaders.py @@ -119,6 +119,8 @@ def generator( # This should be fixed at some point. row = preprocessor(row) # type: ignore[assignment] yield row # type: ignore[misc] + except StopIteration: + raise # Stop iteration when any dataset is exhausted except Exception as err: # noqa: BLE001 # Exception logged logger.error(f"Skipping data row due to error: {err}") gen_count -= 1