This commit is contained in:
Pepijn
2026-03-18 10:15:16 -07:00
parent 219c08ccb8
commit dad661012d
@@ -14,7 +14,6 @@ import pandas as pd
from huggingface_hub import snapshot_download
# ─────────────────────── Config ───────────────────────
DATASETS = [
{"repo_id": "lerobot-data-collection/level2_final_quality3", "episode": 1100},
]
@@ -33,8 +32,6 @@ SCORE_FONT_SCALE = 0.8
TASK_FONT_SCALE = 0.55
# ─────────────────────── Helpers ──────────────────────
def download_episode(repo_id: str, episode: int) -> Path:
"""Download only the files needed for this episode."""
safe_ep = f"{episode:06d}"
@@ -410,8 +407,6 @@ def composite_video(
return gif_path
# ─────────────────────── Main ──────────────────────────
def process_dataset(repo_id: str, episode: int):
safe_name = repo_id.replace("/", "_")
print(f"\n{'='*60}")