code: def _pformat_short_array(
      shape: tuple[int, ...], dtype: str, kind: Optional[str]
  ) -> pp.Doc:
      out = pformat_short_array_text(shape, dtype)
      if kind is not None:
          out = out + f"({kind})"
      return pp.text(out)
  
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 9 for calling a global function with example usage: _pformat_short_array(shape, dtype, kind) and returns: pp