fix test examples et model pretrained

2019-07-11 22:29:08 +02:00
parent 92a782b108
commit c6bf1a400d
3 changed files with 4 additions and 2 deletions
--- a/pytorch_transformers/tests/modeling_utils_test.py
+++ b/pytorch_transformers/tests/modeling_utils_test.py
@@ -30,6 +30,7 @@ class ModelUtilsTest(unittest.TestCase):
            self.assertIsNotNone(config)
            self.assertIsInstance(config, PretrainedConfig)

+            model = BertModel.from_pretrained(model_name)
            model, loading_info = BertModel.from_pretrained(model_name, output_loading_info=True)
            self.assertIsNotNone(model)
            self.assertIsInstance(model, PreTrainedModel)