style(imports): simplify soundfile imports

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
+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]
@@ -26,7 +26,7 @@ from typing import Any
import numpy as np
import sounddevice as sd
import soundfile as sf
from soundfile import SoundFile
from lerobot.utils.errors import (
DeviceAlreadyConnectedError,
@@ -196,7 +196,7 @@ class PortAudioMicrophone(Microphone):
Thread/Process-safe loop to write audio data into a file.
"""
# Can only be run on a single process/thread for file writing safety
with sf.SoundFile(
with SoundFile(
output_file,
mode="w",
samplerate=sample_rate,