From 114e09f570f2c512af6e648183dce3221fde0bf5 Mon Sep 17 00:00:00 2001 From: Mishig Date: Fri, 30 Aug 2024 10:53:55 +0200 Subject: [PATCH] rm EpisodeSampler from viz (#389) --- lerobot/scripts/visualize_dataset_html.py | 14 -------------- 1 file changed, 14 deletions(-) diff --git a/lerobot/scripts/visualize_dataset_html.py b/lerobot/scripts/visualize_dataset_html.py index 7048e7a90..c035e5626 100644 --- a/lerobot/scripts/visualize_dataset_html.py +++ b/lerobot/scripts/visualize_dataset_html.py @@ -57,7 +57,6 @@ import logging import shutil from pathlib import Path -import torch import tqdm from flask import Flask, redirect, render_template, url_for @@ -65,19 +64,6 @@ from lerobot.common.datasets.lerobot_dataset import LeRobotDataset from lerobot.common.utils.utils import init_logging -class EpisodeSampler(torch.utils.data.Sampler): - def __init__(self, dataset, episode_index): - from_idx = dataset.episode_data_index["from"][episode_index].item() - to_idx = dataset.episode_data_index["to"][episode_index].item() - self.frame_ids = range(from_idx, to_idx) - - def __iter__(self): - return iter(self.frame_ids) - - def __len__(self): - return len(self.frame_ids) - - def run_server( dataset: LeRobotDataset, episodes: list[int],