Correct slow test

e615269c · Lysandre · Lysandre Debut · 5f96ebc0 · e615269c
Commit e615269c authored Feb 03, 2020 by Lysandre Committed by Lysandre Debut Feb 04, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/test_modeling_flaubert.py tests/test_modeling_flaubert.py +1 -1

No files found.
--- 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)