code: def _pformat_function(obj: types.FunctionType, **kwargs) -> pp.Doc:
      if kwargs.get("wrapped", False):
          fn = "wrapped function"
      else:
          fn = "function"
      return pp.text(f"<{fn} {obj.__name__}>")
  
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 11 for calling a global function with example usage: _pformat_function(obj) and returns: pp