From e615269cb864392b15073f80ce32643b704776a3 Mon Sep 17 00:00:00 2001 From: Lysandre Date: Mon, 3 Feb 2020 18:33:27 -0500 Subject: [PATCH] Correct slow test --- tests/test_modeling_flaubert.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_modeling_flaubert.py b/tests/test_modeling_flaubert.py index 25345106dc..8af1c8d40a 100644 --- a/tests/test_modeling_flaubert.py +++ b/tests/test_modeling_flaubert.py @@ -387,6 +387,6 @@ class FlaubertModelTest(ModelTesterMixin, unittest.TestCase): @slow def test_model_from_pretrained(self): - for model_name in list(Flaubert_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: + for model_name in list(FLAUBERT_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: model = FlaubertModel.from_pretrained(model_name, cache_dir=CACHE_DIR) self.assertIsNotNone(model)