Small fix

HSTEHSTEHSTE · Jan 16, 2024 · 4a4607c · 4a4607c
1 parent abea30d
commit 4a4607c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/estimators/speech_recognition/test_pytorch_icefall.py b/tests/estimators/speech_recognition/test_pytorch_icefall.py
@@ -88,9 +88,9 @@ def test_pytorch_icefall(art_warning, expected_values, device_type):
 
         # Test transcription outputs
         hyps = []
-        hyps.append(speech_recognizer.predict(x[0]))
-        hyps.append(speech_recognizer.predict(x[1]))
-        hyps.append(speech_recognizer.predict(x[2]))
+        hyps.append(speech_recognizer.predict(xs[0]))
+        hyps.append(speech_recognizer.predict(xs[1]))
+        hyps.append(speech_recognizer.predict(xs[2]))
         print(hyps)
 
         assert (np.array(hyps) == y).all()