Skip to content

GitLab

Unverified Commit 83f45cd6 authored by Lysandre Debut's avatar Lysandre Debut Committed by GitHub
Browse files

Fix auto (#15706)

parent d5083c33
Show whitespace changes
Inline Side-by-side
Showing with 15 additions and 25 deletions
tests/test_modeling_tf_auto.py
View file @ 83f45cd6
...@@ -85,12 +85,7 @@ if is_tf_available(): ...@@ -85,12 +85,7 @@ if is_tf_available():
class TFAutoModelTest(unittest.TestCase): class TFAutoModelTest(unittest.TestCase):
@slow @slow
def test_model_from_pretrained(self): def test_model_from_pretrained(self):
import h5py model_name = "bert-base-cased"
self.assertTrue(h5py.version.hdf5_version.startswith("1.10"))
# for model_name in TF_BERT_PRETRAINED_MODEL_ARCHIVE_LIST[:1]:
for model_name in ["bert-base-uncased"]:
config = AutoConfig.from_pretrained(model_name) config = AutoConfig.from_pretrained(model_name)
self.assertIsNotNone(config) self.assertIsNotNone(config)
self.assertIsInstance(config, BertConfig) self.assertIsInstance(config, BertConfig)
...@@ -101,12 +96,7 @@ class TFAutoModelTest(unittest.TestCase): ...@@ -101,12 +96,7 @@ class TFAutoModelTest(unittest.TestCase):
@slow @slow
def test_model_for_pretraining_from_pretrained(self): def test_model_for_pretraining_from_pretrained(self):
import h5py model_name = "bert-base-cased"
self.assertTrue(h5py.version.hdf5_version.startswith("1.10"))
# for model_name in TF_BERT_PRETRAINED_MODEL_ARCHIVE_LIST[:1]:
for model_name in ["bert-base-uncased"]:
config = AutoConfig.from_pretrained(model_name) config = AutoConfig.from_pretrained(model_name)
self.assertIsNotNone(config) self.assertIsNotNone(config)
self.assertIsInstance(config, BertConfig) self.assertIsInstance(config, BertConfig)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment