tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix: small nit, typing of method

Browse Source
This commit is contained in:
fracapuano
2025-05-27 14:32:14 +02:00
parent de5457f538
commit e405f37b95
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/common/datasets/streaming_dataset.py
View File

@@ -182,7 +182,7 @@ class StreamingLeRobotDataset(torch.utils.data.IterableDataset):
self.rng.shuffle(frames_buffer) self.rng.shuffle(frames_buffer)
yield from frames_buffer yield from frames_buffer
def _make_iterable_dataset(self, dataset: datasets.Dataset) -> Iterator: def _make_iterable_dataset(self, dataset: datasets.IterableDataset) -> Iterator:
return iter(dataset) return iter(dataset)
@profile @profile