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,