From e6e54391bd6cf13ae5c232ceff91f6a65573e4b3 Mon Sep 17 00:00:00 2001 From: CarolinePascal Date: Wed, 29 Apr 2026 16:17:40 +0200 Subject: [PATCH] fix(missing method): adding missing `add_microphones_recordings` method in LeRobotDataset --- src/lerobot/datasets/lerobot_dataset.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/src/lerobot/datasets/lerobot_dataset.py b/src/lerobot/datasets/lerobot_dataset.py index 99e9e52f8..011b82362 100644 --- a/src/lerobot/datasets/lerobot_dataset.py +++ b/src/lerobot/datasets/lerobot_dataset.py @@ -386,6 +386,14 @@ class LeRobotDataset(torch.utils.data.Dataset): self._require_writer("add_frame") self.writer.add_frame(frame) + def add_microphones_recordings(self, microphones: dict) -> None: + """Add microphone recordings to the current episode buffer. + + Delegates to :meth:`DatasetWriter.add_microphones_recordings`. + """ + self._require_writer("add_microphones_recordings") + self.writer.add_microphones_recordings(microphones) + def save_episode(self, episode_data: dict | None = None, parallel_encoding: bool = True) -> None: """Save the current episode buffer to disk.