code: def _dict_entry(key: Any, value: Any, **kwargs) -> pp.Doc:
      return pp.concat(
          [
              tree_pp(key, **kwargs),
              pp.text(":"),
              pp.brk(),
              tree_pp(value, **kwargs),
          ]
      )
  
imports: import dataclasses
  import functools
  import sys
  import types
  from collections.abc import Callable, Sequence
  from typing import Any, Optional, Union
  import jax
  import jax._src.pretty_printer
  import jax.tree_util
  import numpy
  from jaxtyping import PyTree
  from ._doc_utils import WithRepr
label_desc: Use image node_id 6 for calling a global function with example usage: _dict_entry(key, value) and returns: pp