From 64ed5258e67f00ae42144cbd58aaf8f85f0c345f Mon Sep 17 00:00:00 2001 From: Simon Alibert Date: Thu, 20 Feb 2025 09:00:14 +0100 Subject: [PATCH] Fix batch convert --- .../common/datasets/v21/batch_convert_dataset_v20_to_v21.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/lerobot/common/datasets/v21/batch_convert_dataset_v20_to_v21.py b/lerobot/common/datasets/v21/batch_convert_dataset_v20_to_v21.py index 624827bd..cee9da16 100644 --- a/lerobot/common/datasets/v21/batch_convert_dataset_v20_to_v21.py +++ b/lerobot/common/datasets/v21/batch_convert_dataset_v20_to_v21.py @@ -29,8 +29,9 @@ LOCAL_DIR = Path("data/") def batch_convert(): status = {} + LOCAL_DIR.mkdir(parents=True, exist_ok=True) logfile = LOCAL_DIR / "conversion_log_v21.txt" - for num, repo_id in available_datasets: + for num, repo_id in enumerate(available_datasets): print(f"\nConverting {repo_id} ({num}/{len(available_datasets)})") print("---------------------------------------------------------") try: