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,