Mark right save_load test as slow (#19031)

huggingface · Sep 14, 2022 · 4eb36f2 · 4eb36f2
1 parent f5f430e
commit 4eb36f2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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 @@ def test_compile_tf_model(self):
 
     # 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 @@ def test_keras_save_load(self):
 
     # 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)