code: def _delete_init_state(x):
      if _is_index(x):
          return tree_at(lambda y: y.init, x, _sentinel)
      else:
          return x
  
imports: import types
  from collections.abc import Callable
  from typing import Any, Generic, TYPE_CHECKING, TypeVar, Union
  from typing_extensions import ParamSpec
  import jax
  import jax.numpy
  import jax.tree_util
  from jaxtyping import PyTree
  from .._module import field, Module
  from .._pretty_print import bracketed, named_objs, text, tree_pformat
  from .._tree import tree_at, tree_equal
label_desc: Use image node_id 2 for calling a global function with example usage: _delete_init_state(x) and returns: tree_at, x