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

Fix slow tests v4.2.0 (#9561)

Browse Source 
* Fix conversational pipeline test

* LayoutLM

* ProphetNet

* BART

* Blenderbot & small

* Marian

* mBART

* Pegasus

* Tapas tokenizer

* BERT2BERT test

* Style

* Example requirements

* TF BERT2BERT test
This commit is contained in:
Lysandre Debut
2021-01-13 09:55:48 -05:00
committed by GitHub
parent 04dc65e5c6
commit c949516695
12 changed files with 39 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
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, slow
from transformers.testing_utils import TestCasePlus, require_datasets, require_torch, slow
if is_datasets_available():
@@ -25,7 +25,13 @@ if is_datasets_available():
class Seq2seqTrainerTester(TestCasePlus):
@slow
@require_datasets
@require_torch
def test_finetune_bert2bert(self):
"""
Currently fails with:
ImportError: To be able to use this metric, you need to install the following dependencies['absl', 'nltk', 'rouge_score']
"""
bert2bert = EncoderDecoderModel.from_encoder_decoder_pretrained("prajjwal1/bert-tiny", "prajjwal1/bert-tiny")
tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")