From 77e6775065e2aebd88ab99e4ef3ddc3569a5acb2 Mon Sep 17 00:00:00 2001
From: Patrick von Platen <patrick.v.platen@gmail.com>
Date: Fri, 6 Mar 2020 15:14:43 +0100
Subject: [PATCH] add current changes

---
 tests/test_modeling_bart.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/test_modeling_bart.py b/tests/test_modeling_bart.py
index 7040c2fd29..d24b996ce5 100644
--- a/tests/test_modeling_bart.py
+++ b/tests/test_modeling_bart.py
@@ -515,7 +515,8 @@ class BartModelIntegrationTest(unittest.TestCase):
             max_length=max_length + 2,
             min_length=min_length + 1,
             no_repeat_ngram_size=3,
-            do_sample=False
+            do_sample=False,
+            early_stopping=True
         )
         decoded = [
             tok.decode(g, skip_special_tokens=True, clean_up_tokenization_spaces=False) for g in hypotheses_batch