ydy0615/lerobot-clone
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-31 10:51:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

test(audio frame): fixing decoded audio frame shape

Browse Source
This commit is contained in:
CarolinePascal
2025-05-20 20:48:57 +02:00
parent 7d430c8067
commit 674f5dfd75
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/datasets/test_datasets.py
View File

@@ -445,7 +445,10 @@ def test_add_frame_audio(audio_dataset):
dataset.save_episode()
assert dataset[0]["audio"].shape == torch.Size(
(int(DEFAULT_AUDIO_CHUNK_DURATION * DEFAULT_SAMPLE_RATE), DUMMY_AUDIO_CHANNELS)
(
DUMMY_AUDIO_CHANNELS,
int(DEFAULT_AUDIO_CHUNK_DURATION * DEFAULT_SAMPLE_RATE),
) # Match pytorch channel-first format
)