diff --git a/lerobot/common/datasets/utils.py b/lerobot/common/datasets/utils.py index e8d85f0ce..9c49db126 100644 --- a/lerobot/common/datasets/utils.py +++ b/lerobot/common/datasets/utils.py @@ -17,11 +17,9 @@ import contextlib import importlib.resources import json import logging -from collections.abc import Iterator from itertools import accumulate from pathlib import Path from pprint import pformat -from types import SimpleNamespace from typing import Any import datasets diff --git a/lerobot/html_dataset_visualizer/src/app/page.tsx b/lerobot/html_dataset_visualizer/src/app/page.tsx index b4029cf36..7cbbcd9a7 100644 --- a/lerobot/html_dataset_visualizer/src/app/page.tsx +++ b/lerobot/html_dataset_visualizer/src/app/page.tsx @@ -15,7 +15,7 @@ export default function Home({ ?.split(/\s+/) .map((x) => parseInt(x.trim(), 10)) .filter((x) => !isNaN(x))[0] ?? 0; - + redirect(`/${process.env.REPO_ID}/episode_${episodeN}`); } // sync with hf.co/spaces URL params diff --git a/lerobot/scripts/visualize_dataset_html.py b/lerobot/scripts/visualize_dataset_html.py index 13f7545dd..3204af643 100644 --- a/lerobot/scripts/visualize_dataset_html.py +++ b/lerobot/scripts/visualize_dataset_html.py @@ -53,19 +53,19 @@ python lerobot/scripts/visualize_dataset_html.py \ """ import argparse -import json -import os -from pathlib import Path -import subprocess import atexit -import signal -import sys +import json import logging +import os +import signal +import subprocess +import sys +from pathlib import Path from flask import Flask, jsonify, redirect, send_file, url_for from lerobot.common.datasets.lerobot_dataset import LeRobotDataset -from lerobot.common.datasets.utils import INFO_PATH, DEFAULT_PARQUET_PATH, DEFAULT_VIDEO_PATH +from lerobot.common.datasets.utils import DEFAULT_PARQUET_PATH, DEFAULT_VIDEO_PATH, INFO_PATH from lerobot.common.utils.utils import init_logging @@ -159,10 +159,10 @@ def build_react_app(script_dir: Path): package_json_path = next_dir / "package.json" build_id_path = next_build_dir / "BUILD_ID" - with open(package_json_path, "r") as f: + with open(package_json_path) as f: package_data = json.load(f) package_version = package_data.get("version", "") - with open(build_id_path, "r") as f: + with open(build_id_path) as f: build_id = f.read().strip() if package_version != build_id: