Fix(tests) fix task index error in test_policies

2026-05-16 17:20:05 +00:00 · 2025-07-06 10:03:19 +02:00
parent 83bf24cc9a
commit bee74c3eab
1 changed files with 5 additions and 1 deletions
@@ -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