update variables

OpenNMT · Aug 15, 2024 · ae69b23 · ae69b23
1 parent 47f38a7
commit ae69b23
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/tests/test_transformers.py b/python/tests/test_transformers.py
@@ -1020,6 +1020,6 @@ def test_transformers_wav2vec2(
 
         predicted_ids = torch.argmax(logits, dim=-1)
         transcription = processor.decode(predicted_ids, output_word_offsets=True)
-        transcriptions = transcription[0].replace(processor.tokenizer.unk_token, "")
+        transcription = transcription[0].replace(processor.tokenizer.unk_token, "")
 
-        assert transcriptions == expected_transcription[0]
+        assert transcription == expected_transcription[0]