Use lru_cache for tokenization tests (#36818)
* fix * fix * fix * fix --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
@@ -17,11 +17,11 @@
|
||||
from transformers import DistilBertTokenizer, DistilBertTokenizerFast
|
||||
from transformers.testing_utils import require_tokenizers, slow
|
||||
|
||||
from ..bert.test_tokenization_bert import BertTokenizationTest
|
||||
from ..bert import test_tokenization_bert
|
||||
|
||||
|
||||
@require_tokenizers
|
||||
class DistilBertTokenizationTest(BertTokenizationTest):
|
||||
class DistilBertTokenizationTest(test_tokenization_bert.BertTokenizationTest):
|
||||
tokenizer_class = DistilBertTokenizer
|
||||
rust_tokenizer_class = DistilBertTokenizerFast
|
||||
test_rust_tokenizer = True
|
||||
|
||||
Reference in New Issue
Block a user