diff --git a/tests/models/bart/test_modeling_bart.py b/tests/models/bart/test_modeling_bart.py index 2570b34cf0eb..637d37984c8f 100644 --- a/tests/models/bart/test_modeling_bart.py +++ b/tests/models/bart/test_modeling_bart.py @@ -962,7 +962,7 @@ def test_xsum_summarization_same_as_fairseq(self): " state." "" ) - dct = tok.batch_encode_plus( + dct = tok( [PGE_ARTICLE], max_length=1024, padding="max_length", @@ -1188,7 +1188,7 @@ def test_cnn_summarization_same_as_fairseq(self): " up to four years in prison. Her next court appearance is scheduled for May 18." ) - dct = tok.batch_encode_plus( + dct = tok( [FRANCE_ARTICLE, SHORTER_ARTICLE, IRAN_ARTICLE, ARTICLE_SUBWAY], max_length=1024, padding="max_length",