Make slow tests slow

This commit is contained in:
Sylvain Gugger
2022-02-09 19:10:22 -05:00
parent c722753afd
commit 644ec05233

View File

@@ -411,6 +411,14 @@ class FNetTokenizationTest(TokenizerTesterMixin, unittest.TestCase):
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,
input_r: dict,