Make slow tests slow

2022-02-09 19:10:22 -05:00
parent c722753afd
commit 644ec05233
1 changed files with 8 additions and 0 deletions
--- a/tests/test_tokenization_fnet.py
+++ b/tests/test_tokenization_fnet.py
@@ -410,6 +410,14 @@ class FNetTokenizationTest(TokenizerTesterMixin, unittest.TestCase):
                input_p = tokenizer_r.pad(input_p, max_length=max_length, padding="max_length")

                self.assert_batch_padded_input_match(input_r, input_p, max_length, pad_token_id)
+    
+    @slow
+    def test_save_pretrained(self):
+        super().test_save_pretrained()
+    
+    @slow
+    def test_save_slow_from_fast_and_reload_fast(self):
+        super().test_save_slow_from_fast_and_reload_fast()

    def assert_batch_padded_input_match(
        self,