Correct model import name in test

huggingface · Jan 30, 2024 · 90107b7 · 90107b7
1 parent 00139d5
commit 90107b7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/xlm_roberta/test_modeling_xlm_roberta.py b/tests/models/xlm_roberta/test_modeling_xlm_roberta.py
@@ -327,7 +327,7 @@ def test_for_multiple_choice(self):
 
     @slow
     def test_model_from_pretrained(self):
-        for model_name in XLMROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST[:1]:
+        for model_name in XLM_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST[:1]:
             model = XLMRobertaModel.from_pretrained(model_name)
             self.assertIsNotNone(model)