Skip to content

Commit 7480ded

Browse files
authored
Fix failing test (huggingface#15104)
1 parent 28e0914 commit 7480ded

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

tests/test_processor_wav2vec2_with_lm.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -243,8 +243,12 @@ def test_decoder_download_ignores_files(self):
243243
path_to_cached_dir = Path(language_model._kenlm_model.path.decode("utf-8")).parent.parent.absolute()
244244

245245
downloaded_decoder_files = os.listdir(path_to_cached_dir)
246+
expected_decoder_files = ["alphabet.json", "language_model"]
247+
248+
downloaded_decoder_files.sort()
249+
expected_decoder_files.sort()
246250

247251
# test that only decoder relevant files from
248252
# https://huggingface.co/hf-internal-testing/processor_with_lm/tree/main
249253
# are downloaded and none of the rest (e.g. README.md, ...)
250-
self.assertListEqual(downloaded_decoder_files, ["alphabet.json", "language_model"])
254+
self.assertListEqual(downloaded_decoder_files, expected_decoder_files)

0 commit comments

Comments
 (0)