From a0c5078df11843268c7d67e5df7f25464c5947cb Mon Sep 17 00:00:00 2001 From: Sai-Suraj-27 Date: Sat, 6 Dec 2025 21:39:47 +0530 Subject: [PATCH] Fixed failing BartModelIntegrationTests --- tests/models/bart/test_modeling_bart.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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",