Skip to content

Commit e4726ab

Browse files
Set default value for line_break to make the test pass. (#1563)
1 parent a676d0a commit e4726ab

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

keras_nlp/models/task_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -64,11 +64,11 @@ def test_summary_with_preprocessor(self):
6464
preprocessor = SimplePreprocessor()
6565
model = SimpleTask(preprocessor)
6666
summary = []
67-
model.summary(print_fn=lambda x, line_break: summary.append(x))
67+
model.summary(print_fn=lambda x, line_break=False: summary.append(x))
6868
self.assertRegex("\n".join(summary), "Preprocessor:")
6969

7070
def test_summary_without_preprocessor(self):
7171
model = SimpleTask()
7272
summary = []
73-
model.summary(print_fn=lambda x, line_break: summary.append(x))
73+
model.summary(print_fn=lambda x, line_break=False: summary.append(x))
7474
self.assertNotRegex("\n".join(summary), "Preprocessor:")

0 commit comments

Comments
 (0)