From 0a6ca58299ca234f6a577e7a4944646112c1a529 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 --- tests/fixtures/dataset_factories.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/fixtures/dataset_factories.py b/tests/fixtures/dataset_factories.py index 84877b8a6..9ff41126e 100644 --- a/tests/fixtures/dataset_factories.py +++ b/tests/fixtures/dataset_factories.py @@ -161,6 +161,7 @@ def info_factory(features_factory): total_frames: int = 0, total_tasks: int = 0, total_videos: int = 0, + total_audio: int = 0, chunks_size: int = DEFAULT_CHUNK_SIZE, data_files_size_in_mb: float = DEFAULT_DATA_FILE_SIZE_IN_MB, video_files_size_in_mb: float = DEFAULT_VIDEO_FILE_SIZE_IN_MB, @@ -180,6 +181,7 @@ def info_factory(features_factory): "total_frames": total_frames, "total_tasks": total_tasks, "total_videos": total_videos, + "total_audio": total_audio, "chunks_size": chunks_size, "data_files_size_in_mb": data_files_size_in_mb, "video_files_size_in_mb": video_files_size_in_mb,