From ae69b23ea34412eaae48c1f30c4fa1a2659d21ac Mon Sep 17 00:00:00 2001 From: hkwon Date: Wed, 14 Aug 2024 18:51:29 -0700 Subject: [PATCH] update variables --- python/tests/test_transformers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/tests/test_transformers.py b/python/tests/test_transformers.py index db00ca228..3c35445fa 100644 --- 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]