Fix accidental edit to test_modeling_common

huggingface · Dec 14, 2021 · 4c99cc7 · 4c99cc7
1 parent a9f7153
commit 4c99cc7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_modeling_common.py b/tests/test_modeling_common.py
@@ -2193,6 +2193,7 @@ def test_push_to_hub(self):
         model = BertModel(config)
         with tempfile.TemporaryDirectory() as tmp_dir:
             model.save_pretrained(os.path.join(tmp_dir, "test-model"), push_to_hub=True, use_auth_token=self._token)
+
             new_model = BertModel.from_pretrained(f"{USER}/test-model")
             for p1, p2 in zip(model.parameters(), new_model.parameters()):
                 self.assertTrue(torch.equal(p1, p2))