make style

huggingface · Mar 17, 2020 · 9857d2e · 9857d2e
1 parent e6e3d28
commit 9857d2e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tests/test_modeling_bart.py b/tests/test_modeling_bart.py
@@ -285,7 +285,12 @@ def test_generate_beam_search(self):
 
         max_length = 5
         new_input_ids = lm_model.generate(
-            input_ids.clone(), do_sample=True, num_return_sequences=1, num_beams=2, no_repeat_ngram_size=3, max_length=max_length
+            input_ids.clone(),
+            do_sample=True,
+            num_return_sequences=1,
+            num_beams=2,
+            no_repeat_ngram_size=3,
+            max_length=max_length,
         )
         self.assertEqual(new_input_ids.shape, (input_ids.shape[0], max_length - 1))
         # TODO(SS): uneven length batches, empty inputs