Skip to content

GitLab

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

Make MLuke tokenizer tests slow (#14690)

parent 13186d71
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
tests/test_tokenization_mluke.py
View file @ e6219320
...@@ -23,6 +23,7 @@ from transformers.testing_utils import require_torch, slow ...@@ -23,6 +23,7 @@ from transformers.testing_utils import require_torch, slow
from .test_tokenization_common import TokenizerTesterMixin from .test_tokenization_common import TokenizerTesterMixin
@slow
class MLukeTokenizerTest(TokenizerTesterMixin, unittest.TestCase): class MLukeTokenizerTest(TokenizerTesterMixin, unittest.TestCase):
tokenizer_class = MLukeTokenizer tokenizer_class = MLukeTokenizer
test_rust_tokenizer = False test_rust_tokenizer = False
...@@ -196,6 +197,7 @@ class MLukeTokenizerTest(TokenizerTesterMixin, unittest.TestCase): ...@@ -196,6 +197,7 @@ class MLukeTokenizerTest(TokenizerTesterMixin, unittest.TestCase):
tokenizer(sentence, entity_spans=[0, 0, 0]) tokenizer(sentence, entity_spans=[0, 0, 0])
@slow
@require_torch @require_torch
class MLukeTokenizerIntegrationTests(unittest.TestCase): class MLukeTokenizerIntegrationTests(unittest.TestCase):
tokenizer_class = MLukeTokenizer tokenizer_class = MLukeTokenizer
......
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