From 7480ded658720267a23a537d30e939ec9218eee0 Mon Sep 17 00:00:00 2001 From: Lysandre Debut Date: Tue, 11 Jan 2022 15:57:34 +0100 Subject: [PATCH] Fix failing test (#15104) --- tests/test_processor_wav2vec2_with_lm.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/test_processor_wav2vec2_with_lm.py b/tests/test_processor_wav2vec2_with_lm.py index 119e8eb7b6..37c6ff01d9 100644 --- a/tests/test_processor_wav2vec2_with_lm.py +++ b/tests/test_processor_wav2vec2_with_lm.py @@ -243,8 +243,12 @@ class Wav2Vec2ProcessorWithLMTest(unittest.TestCase): path_to_cached_dir = Path(language_model._kenlm_model.path.decode("utf-8")).parent.parent.absolute() downloaded_decoder_files = os.listdir(path_to_cached_dir) + expected_decoder_files = ["alphabet.json", "language_model"] + + downloaded_decoder_files.sort() + expected_decoder_files.sort() # test that only decoder relevant files from # https://huggingface.co/hf-internal-testing/processor_with_lm/tree/main # are downloaded and none of the rest (e.g. README.md, ...) - self.assertListEqual(downloaded_decoder_files, ["alphabet.json", "language_model"]) + self.assertListEqual(downloaded_decoder_files, expected_decoder_files)