modeling_bart: 3 small cleanups that dont change outputs (#7381)

* Mbart passing

* boom boom

* cleaner assert

* add assert

* Fix tests
This commit is contained in:
Sam Shleifer
2020-09-25 04:24:14 -04:00
committed by GitHub
parent 9e68d075a4
commit 3c6bf8998f
2 changed files with 9 additions and 5 deletions

View File

@@ -86,8 +86,7 @@ class MBartEnroIntegrationTest(AbstractSeq2SeqIntegrationTest):
batch: BatchEncoding = self.tokenizer.prepare_seq2seq_batch(self.src_text).to(torch_device)
translated_tokens = self.model.generate(**batch)
decoded = self.tokenizer.batch_decode(translated_tokens, skip_special_tokens=True)
self.assertEqual(self.tgt_text[0], decoded[0])
self.assertEqual(self.tgt_text[1], decoded[1])
assert self.tgt_text == decoded
def test_mbart_enro_config(self):
mbart_models = ["facebook/mbart-large-en-ro"]