fix: small nit, typing of method

2025-05-27 14:32:14 +02:00
parent de5457f538
commit e405f37b95
1 changed files with 1 additions and 1 deletions
--- a/lerobot/common/datasets/streaming_dataset.py
+++ b/lerobot/common/datasets/streaming_dataset.py
@@ -182,7 +182,7 @@ class StreamingLeRobotDataset(torch.utils.data.IterableDataset):
        self.rng.shuffle(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)

    @profile