From 9a0399e18d094e17a21521407a781f2f55c394b0 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Mon, 8 Feb 2021 13:25:09 +0300 Subject: [PATCH] fix bart tests (#10060) --- tests/test_modeling_bart.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/tests/test_modeling_bart.py b/tests/test_modeling_bart.py index 2e15fa77d8..33ccbfaa80 100644 --- a/tests/test_modeling_bart.py +++ b/tests/test_modeling_bart.py @@ -42,7 +42,6 @@ if is_torch_available(): BartForSequenceClassification, BartModel, BartTokenizer, - BartTokenizerFast, pipeline, ) from transformers.models.bart.modeling_bart import BartDecoder, BartEncoder, shift_tokens_right @@ -566,10 +565,6 @@ class BartModelIntegrationTests(unittest.TestCase): def default_tokenizer(self): return BartTokenizer.from_pretrained("facebook/bart-large") - @cached_property - def default_tokenizer_fast(self): - return BartTokenizerFast.from_pretrained("facebook/bart-large") - @slow def test_inference_no_head(self): model = BartModel.from_pretrained("facebook/bart-large").to(torch_device) @@ -589,14 +584,14 @@ class BartModelIntegrationTests(unittest.TestCase): pbase = pipeline(task="fill-mask", model="facebook/bart-base") src_text = [" I went to the ."] results = [x["token_str"] for x in pbase(src_text)] - assert "Ġbathroom" in results + assert " bathroom" in results @slow def test_large_mask_filling(self): plarge = pipeline(task="fill-mask", model="facebook/bart-large") src_text = [" I went to the ."] results = [x["token_str"] for x in plarge(src_text)] - expected_results = ["Ġbathroom", "Ġgym", "Ġwrong", "Ġmovies", "Ġhospital"] + expected_results = [" bathroom", " gym", " wrong", " movies", " hospital"] self.assertListEqual(results, expected_results) @slow