admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-21 19:49:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: catch KeyboardInterrupt in safe_stop_image_writer to prevent corrupted frames (#3381)

Browse Source
This commit is contained in:
Khalil Meftah
2026-04-14 18:22:56 +02:00
committed by GitHub
parent 1ede000bdd
commit 60e7d67cb8
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/datasets/image_writer.py
+2 -2
View File
@@ -30,13 +30,13 @@ def safe_stop_image_writer(func):
def wrapper(*args, **kwargs): def wrapper(*args, **kwargs):
try: try:
return func(*args, **kwargs) return func(*args, **kwargs)
except Exception as e: except BaseException:
dataset = kwargs.get("dataset") dataset = kwargs.get("dataset")
writer = getattr(dataset, "writer", None) if dataset else None writer = getattr(dataset, "writer", None) if dataset else None
if writer is not None and writer.image_writer is not None: if writer is not None and writer.image_writer is not None:
logger.warning("Waiting for image writer to terminate...") logger.warning("Waiting for image writer to terminate...")
writer.image_writer.stop() writer.image_writer.stop()
raise e raise
return wrapper return wrapper