From 11cde3bcd4eb015e8fc8a988bafc0af6485cad0b Mon Sep 17 00:00:00 2001
From: Thomas Wolf <thomas@huggingface.co>
Date: Fri, 31 May 2024 11:20:25 +0200
Subject: [PATCH] update push_to_hub

---
 README.md                              |  3 ++-
 lerobot/scripts/push_dataset_to_hub.py | 20 ++++++++++++--------
 2 files changed, 14 insertions(+), 9 deletions(-)

diff --git a/README.md b/README.md
index 12ebe8d0..dc9dd297 100644
--- a/README.md
+++ b/README.md
@@ -231,7 +231,8 @@ huggingface-cli login --token ${HUGGINGFACE_TOKEN} --add-to-git-credential
 Then move your dataset folder in `data` directory (e.g. `data/aloha_static_pingpong_test`), and push your dataset to the hub with:
 ```bash
 python lerobot/scripts/push_dataset_to_hub.py \
---data-dir data \
+--raw-dir data_raw \
+--out-dir out_data \
 --dataset-id aloha_static_pingpong_test \
 --raw-format aloha_hdf5 \
 --community-id lerobot
diff --git a/lerobot/scripts/push_dataset_to_hub.py b/lerobot/scripts/push_dataset_to_hub.py
index 92400249..91ac592a 100644
--- a/lerobot/scripts/push_dataset_to_hub.py
+++ b/lerobot/scripts/push_dataset_to_hub.py
@@ -21,7 +21,8 @@ installation of neural net specific packages like pytorch, tensorflow, jax.
 Example:
 ```
 python lerobot/scripts/push_dataset_to_hub.py \
---data-dir data \
+--raw-data data_raw \
+--out-dir data \
 --dataset-id pusht \
 --raw-format pusht_zarr \
 --community-id lerobot \
@@ -31,7 +32,8 @@ python lerobot/scripts/push_dataset_to_hub.py \
 --debug 1
 
 python lerobot/scripts/push_dataset_to_hub.py \
---data-dir data \
+--raw-dir data_raw \
+--out-dir data \
 --dataset-id xarm_lift_medium \
 --raw-format xarm_pkl \
 --community-id lerobot \
@@ -41,7 +43,8 @@ python lerobot/scripts/push_dataset_to_hub.py \
 --debug 1
 
 python lerobot/scripts/push_dataset_to_hub.py \
---data-dir data \
+--raw-dir data_raw \
+--out-dir data \
 --dataset-id aloha_sim_insertion_scripted \
 --raw-format aloha_hdf5 \
 --community-id lerobot \
@@ -51,7 +54,8 @@ python lerobot/scripts/push_dataset_to_hub.py \
 --debug 1
 
 python lerobot/scripts/push_dataset_to_hub.py \
---data-dir data \
+--raw-dir data_raw \
+--out-dir data \
 --dataset-id umi_cup_in_the_wild \
 --raw-format umi_zarr \
 --community-id lerobot \
@@ -287,10 +291,10 @@ def main():
         "--save-to-disk",
         type=int,
         default=1,
-        help="Save the dataset in the directory specified by `--data-dir`.",
+        help="Save the dataset in the directory specified by `--out-dir`.",
     )
     parser.add_argument(
-        "--tests-data-dir",
+        "--tests-out-dir",
         type=Path,
         default="tests/data",
         help="Directory containing tests artifacts datasets.",
@@ -298,8 +302,8 @@ def main():
     parser.add_argument(
         "--save-tests-to-disk",
         type=int,
-        default=1,
-        help="Save the dataset with 1 episode used for unit tests in the directory specified by `--tests-data-dir`.",
+        default=0,
+        help="Save the dataset with 1 episode used for unit tests in the directory specified by `--tests-out-dir`.",
     )
     parser.add_argument(
         "--fps",