From ca88603c4b58ceb787b8417f5320465194729a98 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Thu, 3 Apr 2025 03:55:07 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- lerobot/scripts/allowed_hosts.py | 6 +-- lerobot/scripts/visualize_dataset_html.py | 64 +++++++++++------------ 2 files changed, 34 insertions(+), 36 deletions(-) diff --git a/lerobot/scripts/allowed_hosts.py b/lerobot/scripts/allowed_hosts.py index 7e7e9a8c..74429b28 100644 --- a/lerobot/scripts/allowed_hosts.py +++ b/lerobot/scripts/allowed_hosts.py @@ -1,7 +1,7 @@ # List of allowed schemes and hosts for external requests -ALLOWED_SCHEMES = {'http', 'https'} +ALLOWED_SCHEMES = {"http", "https"} ALLOWED_HOSTS = { - 'localhost', - '127.0.0.1', + "localhost", + "127.0.0.1", # Add other trusted hosts here as needed } diff --git a/lerobot/scripts/visualize_dataset_html.py b/lerobot/scripts/visualize_dataset_html.py index d6b12a27..306461c2 100644 --- a/lerobot/scripts/visualize_dataset_html.py +++ b/lerobot/scripts/visualize_dataset_html.py @@ -15,29 +15,25 @@ import argparse import base64 -import json import os import sys import tempfile import urllib.parse from pathlib import Path -from typing import Dict, List, Optional, Tuple, Union +from typing import Dict, List, Tuple, Union import cv2 -import flask import numpy as np import requests -from flask import Flask, Response, jsonify, request +from allowed_hosts import ALLOWED_HOSTS, ALLOWED_SCHEMES +from flask import Flask, jsonify, request from flask_cors import CORS -from PIL import Image from lerobot.data.dataset import Dataset from lerobot.data.episode import Episode from lerobot.data.frame import Frame from lerobot.data.utils import get_dataset_path -from allowed_hosts import ALLOWED_SCHEMES, ALLOWED_HOSTS - app = Flask(__name__) CORS(app) @@ -45,15 +41,15 @@ CORS(app) def validate_url(url): """Validate URL against allowed schemes and hosts.""" parsed_url = urllib.parse.urlparse(url) - + # Check if scheme is allowed if parsed_url.scheme not in ALLOWED_SCHEMES: return False - + # Check if host is allowed if parsed_url.netloc not in ALLOWED_HOSTS: return False - + return True @@ -203,52 +199,52 @@ def index():

Dataset Viewer

- +
- +
Frame: 0 / 0
- +

RGB Image

RGB Image
- +

Depth Image

Depth Image
- +

Frame Information


             

-            
+
             

                 
                 
             

         

-        
+