code: def bracketed(
      name: Optional[pp.Doc],
      indent: int,
      objs: Sequence[pp.Doc],
      lbracket: str,
      rbracket: str,
  ) -> pp.Doc:
      nested = pp.concat(
          [
              pp.nest(
                  indent,
                  pp.concat([pp.brk(""), pp.join(_comma_sep, objs)]),
              ),
              pp.brk(""),
          ]
      )
      concated = []
      if name is not None:
          concated.append(name)
      concated.extend([pp.text(lbracket), nested, pp.text(rbracket)])
      return pp.group(pp.concat(concated))
  
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 1 for calling a global function with example usage: bracketed(name, indent, objs, lbracket, rbracket) and returns: pp