tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Fix batch convert

Browse Source
This commit is contained in:
Simon Alibert
2025-02-20 09:00:14 +01:00
parent 392a8c32a7
commit 64ed5258e6
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
lerobot/common/datasets/v21/batch_convert_dataset_v20_to_v21.py
View File

@@ -29,8 +29,9 @@ LOCAL_DIR = Path("data/")
def batch_convert(): def batch_convert():
status = {} status = {}
LOCAL_DIR.mkdir(parents=True, exist_ok=True)
logfile = LOCAL_DIR / "conversion_log_v21.txt" 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(f"\nConverting {repo_id} ({num}/{len(available_datasets)})")
print("---------------------------------------------------------") print("---------------------------------------------------------")
try: try: