admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-28 06:59:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix(tests) fix task index error in test_policies

Browse Source
This commit is contained in:
Michel Aractingi
2025-07-06 10:03:19 +02:00
parent 83bf24cc9a
commit bee74c3eab
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/policies/test_policies.py
+5 -1
View File
@@ -69,7 +69,11 @@ def dummy_dataset_metadata(lerobot_dataset_metadata_factory, info_factory, tmp_p
}, },
} }
info = info_factory( info = info_factory(
total_episodes=1, total_frames=1, camera_features=camera_features, motor_features=motor_features total_episodes=1,
total_frames=1,
total_tasks=1,
camera_features=camera_features,
motor_features=motor_features,
) )
ds_meta = lerobot_dataset_metadata_factory(root=tmp_path / "init", info=info) ds_meta = lerobot_dataset_metadata_factory(root=tmp_path / "init", info=info)
return ds_meta return ds_meta