From 176e08e6f0c54f6e0c75da173726d65548faf5b8 Mon Sep 17 00:00:00 2001 From: lial1115 Date: Wed, 27 Aug 2025 11:43:27 +0800 Subject: [PATCH] Fix convert_to_lerobot script import path and API call issues to adapt lerobot v0.3.3 - Fix LeRobotDataset import path from lerobot.common.datasets.lerobot_dataset to lerobot.datasets.lerobot_dataset - Fix dataset.add_frame() method call to pass task as second parameter instead of adding to frame dict --- scripts/convert_to_lerobot.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/scripts/convert_to_lerobot.py b/scripts/convert_to_lerobot.py index d15ddbb..e545820 100644 --- a/scripts/convert_to_lerobot.py +++ b/scripts/convert_to_lerobot.py @@ -18,7 +18,7 @@ import json from collections import OrderedDict from pathlib import Path -from lerobot.common.datasets.lerobot_dataset import LeRobotDataset +from lerobot.datasets.lerobot_dataset import LeRobotDataset import numpy as np import PIL @@ -233,8 +233,7 @@ def write_frames( img = np.stack([img] * 3, axis=-1) # reshape from (h,w) to (h,w,3) frame[feature] = img - frame["task"] = task - dataset.add_frame(frame) + dataset.add_frame(frame, task) # Convert a collection of MobilityGen datasets into a single LeRobot dataset. def convert_to_lerobot_dataset(