diff --git a/keras_core/testing/test_case.py b/keras_core/testing/test_case.py index 2bb54e05a..a39a2b371 100644 --- a/keras_core/testing/test_case.py +++ b/keras_core/testing/test_case.py @@ -261,10 +261,8 @@ def run_output_asserts(layer, output, eager=False): self.assertIsInstance(output, list) self.assertEqual( len(output), - len( - expected_output_shape, - msg="Unexpected number of outputs", - ), + len(expected_output_shape), + msg="Unexpected number of outputs", ) output_shape = [v.shape for v in expected_output_shape] self.assertEqual(