fix: fixing typos

This commit is contained in:
CarolinePascal 2025-04-11 18:51:58 +02:00
parent 5267829b5a
commit 89697d86e7
No known key found for this signature in database
2 changed files with 3 additions and 3 deletions

View File

@ -64,12 +64,12 @@ def decode_audio(
if backend == "torchcodec":
raise NotImplementedError("torchcodec is not yet supported for audio decoding")
elif backend == "ffmpeg":
return decode_audio_torchvision(audio_path, timestamps, duration)
return decode_audio_torchaudio(audio_path, timestamps, duration)
else:
raise ValueError(f"Unsupported video backend: {backend}")
def decode_audio_torchvision(
def decode_audio_torchaudio(
audio_path: Path | str,
timestamps: list[float],
duration: float,

View File

@ -87,7 +87,7 @@ def test_sample_audio_from_path(mock_load):
assert len(audio_samples) == estimate_num_samples(16000)
def test_sample_audio_from_data(mock_load):
def test_sample_audio_from_data():
audio_data = np.ones((16000, 2), dtype=np.float32)
audio_samples = sample_audio_from_data(audio_data)
assert isinstance(audio_samples, np.ndarray)