Spaces:
Running
Running
File size: 834 Bytes
d646dd7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 | #!/usr/bin/env python3
"""Repack a project into LeRobot v2.1's 1000-episode chunks."""
from __future__ import annotations
import argparse
import json
import sys
from pathlib import Path
sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
def main() -> int:
parser = argparse.ArgumentParser()
parser.add_argument("project_root", type=Path)
parser.add_argument("--drop-backup", action="store_true",
help="remove the pre-repack tree after verification")
args = parser.parse_args()
from app.project_dataset import repack_project_chunks
result = repack_project_chunks(
args.project_root,
keep_backup=not args.drop_backup,
)
print(json.dumps(result, ensure_ascii=False, indent=2))
return 0
if __name__ == "__main__":
raise SystemExit(main())
|