Typo fix

2022-09-14 10:02:14 -04:00
parent e1224a2a0f
commit 1207deb806
1 changed files with 1 additions and 1 deletions
--- 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,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()
+    @slow
    def test_keras_save_load(self):
        # make mask reproducible
        np.random.seed(2)