Skip to content

Commit

Permalink
Improve doc
Browse files Browse the repository at this point in the history
  • Loading branch information
aliberts committed Feb 20, 2025
1 parent 969ef74 commit 392a8c3
Showing 1 changed file with 7 additions and 6 deletions.
13 changes: 7 additions & 6 deletions lerobot/common/datasets/v21/convert_dataset_v20_to_v21.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,10 +2,10 @@
This script will help you convert any LeRobot dataset already pushed to the hub from codebase version 2.0 to
2.1. It will:
- Generates per-episodes stats and writes them in `episodes_stats.jsonl`
- Generate per-episodes stats and writes them in `episodes_stats.jsonl`
- Check consistency between these new stats and the old ones.
- Removes the deprecated `stats.json` (by default)
- Updates codebase_version in `info.json`
- Remove the deprecated `stats.json`.
- Update codebase_version in `info.json`.
- Push this new version to the hub on the 'main' branch and tags it with "v2.1".
Usage:
Expand Down Expand Up @@ -80,19 +80,20 @@ def convert_dataset(
"--repo-id",
type=str,
required=True,
help="Repository identifier on Hugging Face: a community or a user name `/` the name of the dataset (e.g. `lerobot/pusht`, `cadene/aloha_sim_insertion_human`).",
help="Repository identifier on Hugging Face: a community or a user name `/` the name of the dataset "
"(e.g. `lerobot/pusht`, `cadene/aloha_sim_insertion_human`).",
)
parser.add_argument(
"--branch",
type=str,
default=None,
help="Repo branch to push your dataset (defaults to the main branch)",
help="Repo branch to push your dataset. Defaults to the main branch.",
)
parser.add_argument(
"--num-workers",
type=int,
default=4,
help="Number of workers for parallelizing compute",
help="Number of workers for parallelizing stats compute. Defaults to 4.",
)

args = parser.parse_args()
Expand Down

0 comments on commit 392a8c3

Please # to comment.