From 0d158e38c9e150e2726675666baa531a577da0ef Mon Sep 17 00:00:00 2001 From: Sam Shleifer Date: Mon, 22 Jun 2020 23:31:36 -0400 Subject: [PATCH] [fix] mobilebert had wrong path, causing slow test failure (#5205) --- src/transformers/modeling_mobilebert.py | 2 +- tests/test_modeling_mobilebert.py | 6 ------ 2 files changed, 1 insertion(+), 7 deletions(-) diff --git a/src/transformers/modeling_mobilebert.py b/src/transformers/modeling_mobilebert.py index 018ee4eb40..fa91d77b72 100644 --- a/src/transformers/modeling_mobilebert.py +++ b/src/transformers/modeling_mobilebert.py @@ -39,7 +39,7 @@ from .modeling_utils import PreTrainedModel, find_pruneable_heads_and_indices, p logger = logging.getLogger(__name__) -MOBILEBERT_PRETRAINED_MODEL_ARCHIVE_LIST = ["mobilebert-uncased"] +MOBILEBERT_PRETRAINED_MODEL_ARCHIVE_LIST = ["google/mobilebert-uncased"] def load_tf_weights_in_mobilebert(model, config, tf_checkpoint_path): diff --git a/tests/test_modeling_mobilebert.py b/tests/test_modeling_mobilebert.py index 4d2c934d9d..ee9902c20f 100644 --- a/tests/test_modeling_mobilebert.py +++ b/tests/test_modeling_mobilebert.py @@ -454,12 +454,6 @@ class MobileBertModelTest(ModelTesterMixin, unittest.TestCase): config_and_inputs = self.model_tester.prepare_config_and_inputs() self.model_tester.create_and_check_mobilebert_for_token_classification(*config_and_inputs) - @slow - def test_model_from_pretrained(self): - for model_name in MOBILEBERT_PRETRAINED_MODEL_ARCHIVE_LIST[:1]: - model = MobileBertModel.from_pretrained(model_name) - self.assertIsNotNone(model) - def _long_tensor(tok_lst): return torch.tensor(tok_lst, dtype=torch.long, device=torch_device,)