Mark right save_load test as slow (#19031)

4eb36f29 · Sylvain Gugger · GitHub · f5f430e5 · 4eb36f29
Unverified Commit 4eb36f29 authored Sep 14, 2022 by Sylvain Gugger Committed by GitHub Sep 14, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/models/vit_mae/test_modeling_tf_vit_mae.py tests/models/vit_mae/test_modeling_tf_vit_mae.py +1 -1

No files found.
--- a/tests/models/vit_mae/test_modeling_tf_vit_mae.py
+++ b/tests/models/vit_mae/test_modeling_tf_vit_mae.py
@@ -328,7 +328,6 @@ class TFViTMAEModelTest(TFModelTesterMixin, unittest.TestCase):

    # overwrite from common since TFViTMAEForPretraining has random masking, we need to fix the noise
    # to generate masks during test
-    @slow
    def test_keras_save_load(self):
        # make mask reproducible
        np.random.seed(2)
@@ -376,6 +375,7 @@ class TFViTMAEModelTest(TFModelTesterMixin, unittest.TestCase):

    # overwrite from common since TFViTMAEForPretraining has random masking, we need to fix the noise
    # to generate masks during test
+    @slow
    def test_save_load(self):
        # make mask reproducible
        np.random.seed(2)