From e6219320b91830ed03c315de6ceb5ff55a7b853e Mon Sep 17 00:00:00 2001 From: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> Date: Wed, 8 Dec 2021 15:59:57 -0500 Subject: [PATCH] Make MLuke tokenizer tests slow (#14690) --- tests/test_tokenization_mluke.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/test_tokenization_mluke.py b/tests/test_tokenization_mluke.py index f869bc3292..d2ab3a3f1a 100644 --- a/tests/test_tokenization_mluke.py +++ b/tests/test_tokenization_mluke.py @@ -23,6 +23,7 @@ from transformers.testing_utils import require_torch, slow from .test_tokenization_common import TokenizerTesterMixin +@slow class MLukeTokenizerTest(TokenizerTesterMixin, unittest.TestCase): tokenizer_class = MLukeTokenizer test_rust_tokenizer = False @@ -196,6 +197,7 @@ class MLukeTokenizerTest(TokenizerTesterMixin, unittest.TestCase): tokenizer(sentence, entity_spans=[0, 0, 0]) +@slow @require_torch class MLukeTokenizerIntegrationTests(unittest.TestCase): tokenizer_class = MLukeTokenizer