admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-22 20:19:43 +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 matplotlib.pyplot as plt
import numpy as np import numpy as np
import soundfile as sf from soundfile import read
from lerobot.microphones.configs import MicrophoneConfig from lerobot.microphones.configs import MicrophoneConfig
from lerobot.microphones.portaudio import PortAudioMicrophone, PortAudioMicrophoneConfig from lerobot.microphones.portaudio import PortAudioMicrophone, PortAudioMicrophoneConfig
@@ -97,8 +97,8 @@ def main(
recorded_audio_chunks = all_audio_chunks[i][microphone_key] recorded_audio_chunks = all_audio_chunks[i][microphone_key]
# Load recorded file # Load recorded file
recorded_data, _ = sf.read(recording_dir / f"{microphone_key}_recording_{i}.wav") recorded_data, _ = read(recording_dir / f"{microphone_key}_recording_{i}.wav")
if len(recorded_data.shape) == 1: if recorded_data.ndim == 1:
recorded_data = np.expand_dims(recorded_data, axis=1) recorded_data = np.expand_dims(recorded_data, axis=1)
record_length[i, j] = recorded_data.shape[0] record_length[i, j] = recorded_data.shape[0]
src/lerobot/microphones/portaudio/microphone_portaudio.py
+2 -2
View File
@@ -26,7 +26,7 @@ from typing import Any
import numpy as np import numpy as np
import sounddevice as sd import sounddevice as sd
import soundfile as sf from soundfile import SoundFile
from lerobot.utils.errors import ( from lerobot.utils.errors import (
DeviceAlreadyConnectedError, DeviceAlreadyConnectedError,
@@ -196,7 +196,7 @@ class PortAudioMicrophone(Microphone):
Thread/Process-safe loop to write audio data into a file. Thread/Process-safe loop to write audio data into a file.
""" """
# Can only be run on a single process/thread for file writing safety # Can only be run on a single process/thread for file writing safety
with sf.SoundFile( with SoundFile(
output_file, output_file,
mode="w", mode="w",
samplerate=sample_rate, samplerate=sample_rate,