adding whitespace

OpenNMT · Aug 14, 2024 · 785a0ad · 785a0ad
1 parent 065b240
commit 785a0ad
Showing 1 changed file with 1 addition and 1 deletion.
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)
-        transcription = transcription[0].replace(processor.tokenizer.unk_token,'')
+        transcription = transcription[0].replace(processor.tokenizer.unk_token, '')
 
         assert transcription == expected_transcription[0]