class_docstr: A helper class to build the summary over the training iterations.
class_docstr_tok: ["A","helper","class","to","build","the","summary","over","the","training","iterations","."]
docstr: Update with the given per-step metrics.
docstr_tok: ["Update","with","the","given","per-step","metrics","."]
code: def update(self, metrics_update, extra_logs_update):
      """Update with the given per-step metrics."""
      self.metrics.append(metrics_update)
      self.extra_logs.append(extra_logs_update)
  
imports: import dataclasses
  import operator
  import re
  from typing import Any, Dict, List, Mapping, Optional, Tuple, Union
  from absl import logging
  from big_vision import utils
  from big_vision.models import vit
  from clu import metric_writers
  import flax
  from flax import jax_utils
  from flax.core import frozen_dict
  import jax
  import jax.numpy
  import ml_collections
  import optax
  from scenic.dataset_lib import dataset_utils
  from scenic.dataset_lib import datasets
  from scenic.projects.knowledge_visual_language.models import constants
  from scenic.projects.t5 import model
  from scenic.train_lib import train_utils
  from scenic.train_lib.train_utils import TrainState
label_desc: Use image node_id 1 for calling the SummaryBuilder obj's underlying member method code with example usage: obj.update(metrics_update, extra_logs_update) without return types