Skip to content

Commit fd77ccc

Browse files
committed
fix based on comment
Signed-off-by: guangli.bao <[email protected]>
1 parent 8c67563 commit fd77ccc

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/guidellm/extras/audio.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,11 @@ def _decode_audio( # noqa: C901, PLR0912
107107
raise ValueError(
108108
f"Audio dict must contain either 'data' or 'url' keys, got {audio}"
109109
)
110-
audio_data = audio["data"] if "data" in audio else audio["url"]
110+
audio_data = audio["data"] if "data" in audio else audio.get("url")
111+
if audio_data is None:
112+
raise ValueError(
113+
f"Audio dict must contain either 'data' or 'url' keys, got {audio}"
114+
)
111115
return _decode_audio(
112116
audio=audio_data,
113117
sample_rate=sample_rate,

0 commit comments

Comments
 (0)