fix: fixing typos
This commit is contained in:
parent
5267829b5a
commit
89697d86e7
|
@ -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,
|
||||
|
|
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue