From 0f50e033e9fa9ab9bc70255b4fb78a0a14bcc2cb Mon Sep 17 00:00:00 2001 From: Thomas Wolf Date: Thu, 30 May 2024 23:24:27 +0200 Subject: [PATCH] update script names --- lerobot/scripts/push_dataset_to_hub.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/lerobot/scripts/push_dataset_to_hub.py b/lerobot/scripts/push_dataset_to_hub.py index c6eac5e9..92400249 100644 --- a/lerobot/scripts/push_dataset_to_hub.py +++ b/lerobot/scripts/push_dataset_to_hub.py @@ -144,7 +144,8 @@ def push_videos_to_hub(repo_id, videos_dir, revision): def push_dataset_to_hub( - data_dir: Path, + raw_dir: Path, + out_dir: Path, dataset_id: str, raw_format: str | None, community_id: str, @@ -161,9 +162,6 @@ def push_dataset_to_hub( ): repo_id = f"{community_id}/{dataset_id}" - raw_dir = data_dir / f"{dataset_id}_raw" - - out_dir = data_dir / repo_id meta_data_dir = out_dir / "meta_data" videos_dir = out_dir / "videos" @@ -245,10 +243,16 @@ def main(): parser = argparse.ArgumentParser() parser.add_argument( - "--data-dir", + "--raw-dir", type=Path, required=True, - help="Root directory containing datasets (e.g. `data` or `tmp/data` or `/tmp/lerobot/data`).", + help="Directory containing input raw datasets (e.g. `data/aloha_mobile_chair_raw` or `data/pusht_raw).", + ) + parser.add_argument( + "--out-dir", + type=Path, + required=True, + help="Root directory containing output dataset (e.g. `data/lerobot/aloha_mobile_chair` or `data/lerobot/pusht`).", ) parser.add_argument( "--dataset-id",