From 5be83fbff67a227abfd59afc12bac4fb11201cdc Mon Sep 17 00:00:00 2001 From: Cadene Date: Thu, 11 Apr 2024 13:46:13 +0000 Subject: [PATCH] fix --- examples/1_visualize_dataset.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/1_visualize_dataset.py b/examples/1_visualize_dataset.py index e9b62ff1..ec70ac27 100644 --- a/examples/1_visualize_dataset.py +++ b/examples/1_visualize_dataset.py @@ -2,14 +2,14 @@ import os from pathlib import Path import lerobot -from lerobot.common.datasets.aloha import AlohaDataset +from lerobot.common.datasets.aloha import PushtDataset from lerobot.scripts.visualize_dataset import render_dataset print(lerobot.available_datasets) # >>> ['aloha_sim_insertion_human', 'aloha_sim_insertion_scripted', 'aloha_sim_transfer_cube_human', 'aloha_sim_transfer_cube_scripted', 'pusht', 'xarm_lift_medium'] # TODO(rcadene): remove DATA_DIR -dataset = AlohaDataset("pusht", root=Path(os.environ.get("DATA_DIR"))) +dataset = PushtDataset("pusht", root=Path(os.environ.get("DATA_DIR"))) video_paths = render_dataset( dataset,