Skip to content

GitLab

Unverified Commit 4eb36f29 authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Mark right save_load test as slow (#19031)

parent f5f430e5
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/models/vit_mae/test_modeling_tf_vit_mae.py
View file @ 4eb36f29
...@@ -328,7 +328,6 @@ class TFViTMAEModelTest(TFModelTesterMixin, unittest.TestCase): ...@@ -328,7 +328,6 @@ class TFViTMAEModelTest(TFModelTesterMixin, unittest.TestCase):
# overwrite from common since TFViTMAEForPretraining has random masking, we need to fix the noise # overwrite from common since TFViTMAEForPretraining has random masking, we need to fix the noise
# to generate masks during test # to generate masks during test
@slow
def test_keras_save_load(self): def test_keras_save_load(self):
# make mask reproducible # make mask reproducible
np.random.seed(2) np.random.seed(2)
...@@ -376,6 +375,7 @@ class TFViTMAEModelTest(TFModelTesterMixin, unittest.TestCase): ...@@ -376,6 +375,7 @@ class TFViTMAEModelTest(TFModelTesterMixin, unittest.TestCase):
# overwrite from common since TFViTMAEForPretraining has random masking, we need to fix the noise # overwrite from common since TFViTMAEForPretraining has random masking, we need to fix the noise
# to generate masks during test # to generate masks during test
@slow
def test_save_load(self): def test_save_load(self):
# make mask reproducible # make mask reproducible
np.random.seed(2) np.random.seed(2)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment