Fix Progbar.update when receiving list, np arrays, and tensors. (#21823)

* Fix `Progbar.update` when encountering list, np array and tensor.

* Resolve comments.

Author: james77777778

keras/src/utils/progbar.py

@@ -3,6 +3,8 @@
 import sys
 import time
 
+import numpy as np
+
 from keras.src.api_export import keras_export
 from keras.src.utils import io_utils
 
@@ -161,7 +163,10 @@ def update(self, current, values=None, finalize=None):
             for k in self._values_order:
                 info += f" - {k}:"
                 if isinstance(self._values[k], list):
-                    avg = self._values[k][0] / max(1, self._values[k][1])
+                    values, count = self._values[k]
+                    if not isinstance(values, float):
+                        values = np.mean(values)
+                    avg = values / max(1, count)
                     if abs(avg) > 1e-3:
                         info += f" {avg:.4f}"
                     else:
@@ -188,7 +193,10 @@ def update(self, current, values=None, finalize=None):
                 info += f" -{self._format_time(time_per_unit, self.unit_name)}"
                 for k in self._values_order:
                     info += f" - {k}:"
-                    avg = self._values[k][0] / max(1, self._values[k][1])
+                    values, count = self._values[k]
+                    if not isinstance(values, float):
+                        values = np.mean(values)
+                    avg = values / max(1, count)
                     if avg > 1e-3:
                         info += f" {avg:.4f}"
                     else:

keras/src/utils/progbar_test.py

@@ -0,0 +1,27 @@
+import numpy as np
+from absl.testing import parameterized
+
+from keras.src import testing
+from keras.src.utils import progbar
+
+
+class ProgbarTest(testing.TestCase):
+    @parameterized.named_parameters(
+        [
+            ("float", "float"),
+            ("np", "np"),
+            ("list", "list"),
+        ]
+    )
+    def test_update(self, value_type):
+        if value_type == "float":
+            values = 1.0
+        elif value_type == "np":
+            values = np.array(1.0)
+        elif value_type == "list":
+            values = [0.0, 1.0, 2.0]
+        else:
+            raise ValueError("Unknown value_type")
+        pb = progbar.Progbar(target=1, verbose=1)
+
+        pb.update(1, values=[("values", values)], finalize=True)