code: def __init__(
      self,
      mapping: tp.Union[
          tp.Mapping[Key, tp.Any],
          tp.Iterator[tp.Tuple[Key, tp.Any]],
      ],
      /,
  ):
      if tp.TYPE_CHECKING:
          self._mapping = dict(mapping)
      else:
          super().__setattr__("_mapping", dict(mapping))
  
imports: from __future__ import annotations
  import typing
  import jax
  import jax.tree_util
  from flax import traverse_util
  from flax.experimental.nnx.nnx import filterlib, reprlib
  from flax.experimental.nnx.nnx.variables import Variable
label_desc: Use image node_id 1 to create a new State object from inherited base classes: tp.MutableMapping, reprlib with example: obj = State()