From f73db4394be5b5b065a5b268d191a6348f7c2754 Mon Sep 17 00:00:00 2001 From: CarolinePascal Date: Thu, 24 Apr 2025 11:54:21 +0200 Subject: [PATCH] fix(audio chunks): querying audio chunks in the past rather than in the future --- src/lerobot/datasets/audio_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/lerobot/datasets/audio_utils.py b/src/lerobot/datasets/audio_utils.py index 1156fa33f..755676737 100644 --- a/src/lerobot/datasets/audio_utils.py +++ b/src/lerobot/datasets/audio_utils.py @@ -76,7 +76,7 @@ def decode_audio_torchcodec( audio_chunks = [] for ts in timestamps: current_audio_chunk = audio_decoder.get_samples_played_in_range( - start_seconds=ts, stop_seconds=ts + duration + start_seconds=ts - duration, stop_seconds=ts ) if log_loaded_timestamps: @@ -114,7 +114,7 @@ def decode_audio_torchaudio( audio_chunks = [] for ts in timestamps: - reader.seek(ts) # Default to closest audio sample + reader.seek(ts - duration) # Default to closest audio sample status = reader.fill_buffer() if status != 0: logging.warning("Audio stream reached end of recording before decoding desired timestamps.")