Make sure your convert your LeRobotDataset to v2 & above.