From c037b2e34041edf9886bbf63caf25d2ac0fa359b Mon Sep 17 00:00:00 2001 From: Arthur <48595927+ArthurZucker@users.noreply.github.com> Date: Wed, 4 Oct 2023 17:47:55 +0200 Subject: [PATCH] skip flaky hub tests (#26594) skip flaky --- tests/test_modeling_utils.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/test_modeling_utils.py b/tests/test_modeling_utils.py index 6d37f2247e..bccde5af50 100755 --- a/tests/test_modeling_utils.py +++ b/tests/test_modeling_utils.py @@ -1096,6 +1096,7 @@ class ModelPushToHubTester(unittest.TestCase): except HTTPError: pass + @unittest.skip("This test is flaky") def test_push_to_hub(self): config = BertConfig( vocab_size=99, hidden_size=32, num_hidden_layers=5, num_attention_heads=4, intermediate_size=37 @@ -1118,6 +1119,7 @@ class ModelPushToHubTester(unittest.TestCase): for p1, p2 in zip(model.parameters(), new_model.parameters()): self.assertTrue(torch.equal(p1, p2)) + @unittest.skip("This test is flaky") def test_push_to_hub_in_organization(self): config = BertConfig( vocab_size=99, hidden_size=32, num_hidden_layers=5, num_attention_heads=4, intermediate_size=37