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_trainer_seq2seq.py
View File

@@ -15,7 +15,7 @@
from transformers import BertTokenizer, EncoderDecoderModel, Seq2SeqTrainer, Seq2SeqTrainingArguments
from transformers.file_utils import is_datasets_available
from transformers.testing_utils import TestCasePlus, require_datasets, require_torch, slow
from transformers.testing_utils import TestCasePlus, require_torch, slow
if is_datasets_available():
@@ -25,7 +25,6 @@ if is_datasets_available():
class Seq2seqTrainerTester(TestCasePlus):
@slow
@require_torch
@require_datasets
def test_finetune_bert2bert(self):
bert2bert = EncoderDecoderModel.from_encoder_decoder_pretrained("prajjwal1/bert-tiny", "prajjwal1/bert-tiny")
tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")