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

Remove datasets requirement (#14795)

Browse Source
This commit is contained in:
Lysandre Debut
2021-12-16 14:34:14 -05:00
committed by GitHub
parent bef1e3e4a0
commit d194d639ab
19 changed files with 11 additions and 65 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/test_tokenization_rag.py
View File

@@ -24,7 +24,7 @@ from transformers.models.bart.configuration_bart import BartConfig
from transformers.models.bert.tokenization_bert import VOCAB_FILES_NAMES as DPR_VOCAB_FILES_NAMES
from transformers.models.dpr.configuration_dpr import DPRConfig
from transformers.models.roberta.tokenization_roberta import VOCAB_FILES_NAMES as BART_VOCAB_FILES_NAMES
from transformers.testing_utils import require_datasets, require_faiss, require_tokenizers, require_torch, slow
from transformers.testing_utils import require_faiss, require_tokenizers, require_torch, slow
if is_torch_available() and is_datasets_available() and is_faiss_available():
@@ -33,7 +33,6 @@ if is_torch_available() and is_datasets_available() and is_faiss_available():
@require_faiss
@require_datasets
@require_torch
class RagTokenizerTest(TestCase):
def setUp(self):