admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-20 11:09:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(missing method): adding missing add_microphones_recordings method in LeRobotDataset

Browse Source
This commit is contained in:
CarolinePascal
2026-04-29 16:17:40 +02:00
parent a201b33d20
commit e6e54391bd
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/datasets/lerobot_dataset.py
+8
View File
@@ -386,6 +386,14 @@ class LeRobotDataset(torch.utils.data.Dataset):
self._require_writer("add_frame") self._require_writer("add_frame")
self.writer.add_frame(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: def save_episode(self, episode_data: dict | None = None, parallel_encoding: bool = True) -> None:
"""Save the current episode buffer to disk. """Save the current episode buffer to disk.