diff --git a/tests/policies/test_policies.py b/tests/policies/test_policies.py index 3243713cb..0add55858 100644 --- a/tests/policies/test_policies.py +++ b/tests/policies/test_policies.py @@ -69,7 +69,11 @@ def dummy_dataset_metadata(lerobot_dataset_metadata_factory, info_factory, tmp_p }, } 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) return ds_meta