From 60e7d67cb80c0be55c3f7375a70bfd12c6571fa3 Mon Sep 17 00:00:00 2001 From: Khalil Meftah Date: Tue, 14 Apr 2026 18:22:56 +0200 Subject: [PATCH] fix: catch KeyboardInterrupt in safe_stop_image_writer to prevent corrupted frames (#3381) --- src/lerobot/datasets/image_writer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/lerobot/datasets/image_writer.py b/src/lerobot/datasets/image_writer.py index 603067757..8fb5804a5 100644 --- a/src/lerobot/datasets/image_writer.py +++ b/src/lerobot/datasets/image_writer.py @@ -30,13 +30,13 @@ def safe_stop_image_writer(func): def wrapper(*args, **kwargs): try: return func(*args, **kwargs) - except Exception as e: + except BaseException: dataset = kwargs.get("dataset") writer = getattr(dataset, "writer", None) if dataset else None if writer is not None and writer.image_writer is not None: logger.warning("Waiting for image writer to terminate...") writer.image_writer.stop() - raise e + raise return wrapper