From c8849583ad966591e9bbfb984d6bf35d89a272e9 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Wed, 18 Jan 2023 10:43:05 +0100 Subject: [PATCH] Make `test_save_pretrained_signatures` slow test (#21105) Co-authored-by: ydshieh --- tests/test_modeling_tf_common.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/test_modeling_tf_common.py b/tests/test_modeling_tf_common.py index f8ca850626..178c50de18 100644 --- a/tests/test_modeling_tf_common.py +++ b/tests/test_modeling_tf_common.py @@ -2266,6 +2266,7 @@ class UtilsFunctionsTest(unittest.TestCase): for p1, p2 in zip(model.weights, new_model.weights): self.assertTrue(np.allclose(p1.numpy(), p2.numpy())) + @slow def test_save_pretrained_signatures(self): model = TFBertModel.from_pretrained("hf-internal-testing/tiny-random-bert")