admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-21 03:30:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(dataset): update imports across the codebase

Browse Source
This commit is contained in:
Steven Palma
2026-03-15 23:09:52 -07:00
parent 26d732c8c8
commit d0ae3e9481
49 changed files with 342 additions and 410 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/datasets/test_sampler.py
+2 -2
View File
@@ -19,10 +19,10 @@ import pytest
import torch
from datasets import Dataset
from lerobot.datasets.sampler import EpisodeAwareSampler
from lerobot.datasets.utils import (
from lerobot.datasets.io_utils import (
hf_transform_to_torch,
)
from lerobot.datasets.sampler import EpisodeAwareSampler
def calculate_episode_data_index(hf_dataset: Dataset) -> dict[str, torch.Tensor]: