This looks like debug code accidentally committed in b18509c2.

Refs #2250.
This commit is contained in:
Aymeric Augustin 2019-12-21 08:50:15 +01:00
Родитель 767bc3ca68
Коммит b8e924e10d
1 изменённых файлов: 1 добавлений и 2 удалений

Просмотреть файл

@ -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)