diff --git a/python/tests/test_transformers.py b/python/tests/test_transformers.py index 5e8c2dba8..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) - transcription = transcription[0].replace(processor.tokenizer.unk_token, '') + transcription = transcription[0].replace(processor.tokenizer.unk_token, "") assert transcription == expected_transcription[0]