Set default value for line_break to make the test pass.

SamanehSaadat · Apr 8, 2024 · da6cfd1 · da6cfd1
1 parent a676d0a
commit da6cfd1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/keras_nlp/models/task_test.py b/keras_nlp/models/task_test.py
@@ -64,11 +64,11 @@ def test_summary_with_preprocessor(self):
         preprocessor = SimplePreprocessor()
         model = SimpleTask(preprocessor)
         summary = []
-        model.summary(print_fn=lambda x, line_break: summary.append(x))
+        model.summary(print_fn=lambda x, line_break=False: summary.append(x))
         self.assertRegex("\n".join(summary), "Preprocessor:")
 
     def test_summary_without_preprocessor(self):
         model = SimpleTask()
         summary = []
-        model.summary(print_fn=lambda x, line_break: summary.append(x))
+        model.summary(print_fn=lambda x, line_break=False: summary.append(x))
         self.assertNotRegex("\n".join(summary), "Preprocessor:")