From da6cfd1f75d6fd467b42a4ba93a3ea918898a874 Mon Sep 17 00:00:00 2001 From: Samaneh Saadat Date: Mon, 8 Apr 2024 23:52:07 +0000 Subject: [PATCH] Set default value for line_break to make the test pass. --- keras_nlp/models/task_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/keras_nlp/models/task_test.py b/keras_nlp/models/task_test.py index d9af5c4902..1b2458d098 100644 --- 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:")