From c7930d617852572cf0e4786952a43a12e338a9c6 Mon Sep 17 00:00:00 2001 From: CarolinePascal Date: Fri, 11 Apr 2025 15:48:23 +0200 Subject: [PATCH] fix: adding proper definition for "total_audio" in LeRobotDataset --- lerobot/common/datasets/utils.py | 1 + tests/fixtures/dataset_factories.py | 2 ++ 2 files changed, 3 insertions(+) diff --git a/lerobot/common/datasets/utils.py b/lerobot/common/datasets/utils.py index 416d5837..7ac46013 100644 --- a/lerobot/common/datasets/utils.py +++ b/lerobot/common/datasets/utils.py @@ -452,6 +452,7 @@ def create_empty_dataset_info( "total_frames": 0, "total_tasks": 0, "total_videos": 0, + "total_audio": 0, "total_chunks": 0, "chunks_size": DEFAULT_CHUNK_SIZE, "fps": fps, diff --git a/tests/fixtures/dataset_factories.py b/tests/fixtures/dataset_factories.py index 321dec46..6de43ef1 100644 --- a/tests/fixtures/dataset_factories.py +++ b/tests/fixtures/dataset_factories.py @@ -121,6 +121,7 @@ def info_factory(features_factory): total_frames: int = 0, total_tasks: int = 0, total_videos: int = 0, + total_audio: int = 0, total_chunks: int = 0, chunks_size: int = DEFAULT_CHUNK_SIZE, data_path: str = DEFAULT_PARQUET_PATH, @@ -139,6 +140,7 @@ def info_factory(features_factory): "total_frames": total_frames, "total_tasks": total_tasks, "total_videos": total_videos, + "total_audio": total_audio, "total_chunks": total_chunks, "chunks_size": chunks_size, "fps": fps,