Unverified Commit 3ea04699 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Removed an outdated check about hdf5_version (#16011)



* removed an outdated check about hdf5_version
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent c1aaa439
......@@ -72,10 +72,6 @@ if is_torch_available():
class TFPTAutoModelTest(unittest.TestCase):
@slow
def test_model_from_pretrained(self):
import h5py
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)
......@@ -92,10 +88,6 @@ class TFPTAutoModelTest(unittest.TestCase):
@slow
def test_model_for_pretraining_from_pretrained(self):
import h5py
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)
......
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