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

fix(typos): fixing typos in audio handling in LeRobotDataset

Browse Source
This commit is contained in:
CarolinePascal
2025-11-02 14:56:37 +01:00
parent 4eb7694d47
commit 9f50913b9c
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/__init__.py
+1
View File
@@ -29,6 +29,7 @@ Example:
print(lerobot.available_policies_per_env) print(lerobot.available_policies_per_env)
print(lerobot.available_robots) print(lerobot.available_robots)
print(lerobot.available_cameras) print(lerobot.available_cameras)
print(lerobot.available_microphones)
print(lerobot.available_motors) print(lerobot.available_motors)
``` ```
src/lerobot/datasets/lerobot_dataset.py
+2 -3
View File
@@ -75,7 +75,6 @@ from lerobot.datasets.video_utils import (
concatenate_media_files, concatenate_media_files,
decode_video_frames, decode_video_frames,
encode_video_frames, encode_video_frames,
get_audio_duration_in_s,
get_media_duration_in_s, get_media_duration_in_s,
get_safe_default_codec, get_safe_default_codec,
get_video_info, get_video_info,
@@ -1756,7 +1755,7 @@ class LeRobotDataset(torch.utils.data.Dataset):
# Encode episode audio into a temporary audio file # Encode episode audio into a temporary audio file
ep_path = self._encode_temporary_episode_audio(audio_key, episode_index) ep_path = self._encode_temporary_episode_audio(audio_key, episode_index)
ep_size_in_mb = get_file_size_in_mb(ep_path) ep_size_in_mb = get_file_size_in_mb(ep_path)
ep_duration_in_s = get_audio_duration_in_s(ep_path) ep_duration_in_s = get_media_duration_in_s(ep_path, media_type="audio")
if ( if (
episode_index == 0 episode_index == 0
@@ -1844,7 +1843,7 @@ class LeRobotDataset(torch.utils.data.Dataset):
if isinstance(episode_index, np.ndarray): if isinstance(episode_index, np.ndarray):
episode_index = episode_index.item() if episode_index.size == 1 else episode_index[0] episode_index = episode_index.item() if episode_index.size == 1 else episode_index[0]
for audio_key in self.meta.audio_keys: for audio_key in self.meta.audio_keys:
audio_file = self.root / self.meta.get_audio_file_path(episode_index, audio_key) audio_file = self._get_raw_audio_file_path(episode_index, audio_key)
if audio_file.is_file(): if audio_file.is_file():
audio_file.unlink() audio_file.unlink()