From bee74c3eab187c64e108f19bb023cc5731c89464 Mon Sep 17 00:00:00 2001 From: Michel Aractingi Date: Sun, 6 Jul 2025 10:03:19 +0200 Subject: [PATCH] Fix(tests) fix task index error in test_policies --- tests/policies/test_policies.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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