Restore test.

This looks like debug code accidentally committed in b18509c2. Refs #2250.

Restore test.
This looks like debug code accidentally committed in b18509c2. Refs #2250.
b8e924e1 · Aymeric Augustin · 767bc3ca · b8e924e1
Commit b8e924e1 authored Dec 21, 2019 by Aymeric Augustin
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

transformers/tests/modeling_tf_albert_test.py transformers/tests/modeling_tf_albert_test.py +1 -2

No files found.
--- a/transformers/tests/modeling_tf_albert_test.py
+++ b/transformers/tests/modeling_tf_albert_test.py
@@ -216,8 +216,7 @@ class TFAlbertModelTest(TFCommonTestCases.TFCommonModelTester):

    @slow
    def test_model_from_pretrained(self):
-        # for model_name in list(TF_ALBERT_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]:
-        for model_name in ['albert-base-uncased']:
+        for model_name in list(TF_ALBERT_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]:
            model = TFAlbertModel.from_pretrained(model_name, cache_dir=CACHE_DIR)
            self.assertIsNotNone(model)