From e405f37b95d7acba0796ffc043d20e6a25dd88a0 Mon Sep 17 00:00:00 2001 From: fracapuano Date: Tue, 27 May 2025 14:32:14 +0200 Subject: [PATCH] fix: small nit, typing of method --- lerobot/common/datasets/streaming_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lerobot/common/datasets/streaming_dataset.py b/lerobot/common/datasets/streaming_dataset.py index d671fee2..0c322c3c 100644 --- 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