SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Use lru_cache for tokenization tests (#36818)

Browse Source 
* fix

* fix

* fix

* fix

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2025-03-28 15:09:35 +01:00
committed by GitHub
parent 3af425d4c6
commit 1fcaad6df9
92 changed files with 1301 additions and 884 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/models/dpr/test_tokenization_dpr.py
View File

@@ -25,11 +25,11 @@ from transformers import (
from transformers.testing_utils import require_tokenizers, slow
from transformers.tokenization_utils_base import BatchEncoding
from ..bert.test_tokenization_bert import BertTokenizationTest
from ..bert import test_tokenization_bert
@require_tokenizers
class DPRContextEncoderTokenizationTest(BertTokenizationTest):
class DPRContextEncoderTokenizationTest(test_tokenization_bert.BertTokenizationTest):
tokenizer_class = DPRContextEncoderTokenizer
rust_tokenizer_class = DPRContextEncoderTokenizerFast
test_rust_tokenizer = True
@@ -37,7 +37,7 @@ class DPRContextEncoderTokenizationTest(BertTokenizationTest):
@require_tokenizers
class DPRQuestionEncoderTokenizationTest(BertTokenizationTest):
class DPRQuestionEncoderTokenizationTest(test_tokenization_bert.BertTokenizationTest):
tokenizer_class = DPRQuestionEncoderTokenizer
rust_tokenizer_class = DPRQuestionEncoderTokenizerFast
test_rust_tokenizer = True
@@ -45,7 +45,7 @@ class DPRQuestionEncoderTokenizationTest(BertTokenizationTest):
@require_tokenizers
class DPRReaderTokenizationTest(BertTokenizationTest):
class DPRReaderTokenizationTest(test_tokenization_bert.BertTokenizationTest):
tokenizer_class = DPRReaderTokenizer
rust_tokenizer_class = DPRReaderTokenizerFast
test_rust_tokenizer = True