diff --git a/examples/dataset/visualization_tools/create_progress_videos.py b/examples/dataset/visualization_tools/create_progress_videos.py index da2cf1a40..93c4b7ee7 100644 --- a/examples/dataset/visualization_tools/create_progress_videos.py +++ b/examples/dataset/visualization_tools/create_progress_videos.py @@ -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}")