Skip to content

Commit b356582

Browse files
authored
Fixed failing Bart-Model Integration Tests (#42676)
Fixed failing BartModelIntegrationTests
1 parent 0e0af80 commit b356582

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/models/bart/test_modeling_bart.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -962,7 +962,7 @@ def test_xsum_summarization_same_as_fairseq(self):
962962
" state."
963963
"</s>"
964964
)
965-
dct = tok.batch_encode_plus(
965+
dct = tok(
966966
[PGE_ARTICLE],
967967
max_length=1024,
968968
padding="max_length",
@@ -1188,7 +1188,7 @@ def test_cnn_summarization_same_as_fairseq(self):
11881188
" up to four years in prison. Her next court appearance is scheduled for May 18."
11891189
)
11901190

1191-
dct = tok.batch_encode_plus(
1191+
dct = tok(
11921192
[FRANCE_ARTICLE, SHORTER_ARTICLE, IRAN_ARTICLE, ARTICLE_SUBWAY],
11931193
max_length=1024,
11941194
padding="max_length",

0 commit comments

Comments
 (0)