Simplify the code

README.md

@@ -38,18 +38,15 @@ from vllm import LLM, SamplingParams
 
 
 def wav_bytes_to_numpy(wav_bytes):
-    with io.BytesIO(wav_bytes) as wf:
-        with wave.open(wf, "rb") as wr:
-            if wr.getnchannels() != 1:
-                raise ValueError(f"num_channels must be 1, got {wr.getnchannels()}")
-            if wr.getsampwidth() != 2:
-                raise ValueError(f"sample_width must be 2, got {wr.getsampwidth()}")
-
-            audio_data = wr.readframes(wr.getnframes())
-
-            return (
-                np.frombuffer(audio_data, dtype=np.int16).astype(np.float32) / 32768.0
-            )
+    with wave.open(io.BytesIO(wav_bytes), "rb") as wr:
+        if (num_channels := wr.getnchannels()) != 1:
+            raise ValueError(f"num_channels must be 1, got {num_channels}")
+        if (sample_width := wr.getsampwidth()) != 2:
+            raise ValueError(f"sample_width must be 2, got {sample_width}")
+
+        audio_data = wr.readframes(wr.getnframes())
+
+        return np.frombuffer(audio_data, dtype=np.int16).astype(np.float32) / 32768.0
 
 
 llm = LLM(