diff --git a/examples/aloha_real/convert_aloha_data_to_lerobot.py b/examples/aloha_real/convert_aloha_data_to_lerobot.py index a3a8ddcb24..8e32fd971e 100644 --- a/examples/aloha_real/convert_aloha_data_to_lerobot.py +++ b/examples/aloha_real/convert_aloha_data_to_lerobot.py @@ -10,7 +10,7 @@ from typing import Literal import h5py -from lerobot.common.datasets.lerobot_dataset import LEROBOT_HOME +from lerobot.common.datasets.lerobot_dataset import HF_LEROBOT_HOME from lerobot.common.datasets.lerobot_dataset import LeRobotDataset from lerobot.common.datasets.push_dataset_to_hub._download_raw import download_raw import numpy as np @@ -109,8 +109,8 @@ def create_empty_dataset( ], } - if Path(LEROBOT_HOME / repo_id).exists(): - shutil.rmtree(LEROBOT_HOME / repo_id) + if Path(HF_LEROBOT_HOME / repo_id).exists(): + shutil.rmtree(HF_LEROBOT_HOME / repo_id) return LeRobotDataset.create( repo_id=repo_id, @@ -238,8 +238,8 @@ def port_aloha( mode: Literal["video", "image"] = "image", dataset_config: DatasetConfig = DEFAULT_DATASET_CONFIG, ): - if (LEROBOT_HOME / repo_id).exists(): - shutil.rmtree(LEROBOT_HOME / repo_id) + if (HF_LEROBOT_HOME / repo_id).exists(): + shutil.rmtree(HF_LEROBOT_HOME / repo_id) if not raw_dir.exists(): if raw_repo_id is None: diff --git a/examples/droid/convert_droid_data_to_lerobot.py b/examples/droid/convert_droid_data_to_lerobot.py index 6f80834cff..35a358e4f2 100644 --- a/examples/droid/convert_droid_data_to_lerobot.py +++ b/examples/droid/convert_droid_data_to_lerobot.py @@ -7,7 +7,7 @@ If you want to push your dataset to the Hugging Face Hub, you can use the following command: uv run examples/droid/convert_droid_data_to_lerobot.py --data_dir /path/to/your/data --push_to_hub -The resulting dataset will get saved to the $LEROBOT_HOME directory. +The resulting dataset will get saved to the $HF_LEROBOT_HOME directory. """ from collections import defaultdict