Minor fix

keras-team · Aug 30, 2023 · e63e94c · e63e94c
1 parent 7c7ed95
commit e63e94c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/keras_core/testing/test_case.py b/keras_core/testing/test_case.py
@@ -253,9 +253,9 @@ def run_output_asserts(layer, output, eager=False):
                 elif isinstance(expected_output_shape, list):
                     self.assertTrue(isinstance(output, list))
                     self.assertEqual(
-                        len(output.keys()),
+                        len(output),
                         len(
-                            expected_output_shape.keys(),
+                            expected_output_shape,
                             msg="Unexpected number of outputs",
                         ),
                     )