code: def _pretty_print(self, context: core.JaxprPpContext) -> pp.Doc:
      if isinstance(self.input_index, core.Var):
          index_text = pp.text(core.pp_var(self.input_index, context))
      else:
          index_text = pp.text(self.input_index)
      return pp.concat(
          [
              pp.color(
                  pp.text(self.name), foreground=_ref_effect_color
              ),
              pp.text("<"),
              index_text,
              pp.text(">"),
          ]
      )
  
imports: from __future__ import annotations
  from collections.abc import Sequence
  import dataclasses
  import math
  from typing import Any, Generic, TypeVar, Union
  from jax._src import core
  from jax._src import effects
  from jax._src import pretty_printer
  from jax._src.state import indexing
  from jax._src.util import safe_map, safe_zip
  from jax._src.typing import Array
label_desc: Use image node_id 3 for calling the RefEffect obj's underlying member method code with example usage: obj._pretty_print(context) and returns: pp