wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: adding proper definition for "total_audio" in LeRobotDataset

This commit is contained in:
CarolinePascal 2025-04-11 15:48:23 +02:00
parent 8bea50ecdd
commit c7930d6178
No known key found for this signature in database
2 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
lerobot/common/datasets/utils.py
View File

@ -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,

2
tests/fixtures/dataset_factories.py vendored
View File

@ -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,