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

style(imports): simplify soundfile imports

Browse Source
This commit is contained in:
CarolinePascal
2025-04-22 18:31:08 +02:00
parent 067993bb11
commit 2962330bb1
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
benchmarks/audio/run_microphone_benchmark.py
+3 -3
View File
@@ -20,7 +20,7 @@ from pathlib import Path
import matplotlib.pyplot as plt
import numpy as np
import soundfile as sf
from soundfile import read
from lerobot.microphones.configs import MicrophoneConfig
from lerobot.microphones.portaudio import PortAudioMicrophone, PortAudioMicrophoneConfig
@@ -97,8 +97,8 @@ def main(
recorded_audio_chunks = all_audio_chunks[i][microphone_key]
# Load recorded file
recorded_data, _ = sf.read(recording_dir / f"{microphone_key}_recording_{i}.wav")
if len(recorded_data.shape) == 1:
recorded_data, _ = read(recording_dir / f"{microphone_key}_recording_{i}.wav")
if recorded_data.ndim == 1:
recorded_data = np.expand_dims(recorded_data, axis=1)
record_length[i, j] = recorded_data.shape[0]