code: def _add_next_action_extras(
      double_transitions: Transition,
  ) -> Transition:
      return Transition(
          observation=double_transitions.observation[0],
          action=double_transitions.action[0],
          reward=double_transitions.reward[0],
          discount=double_transitions.discount[0],
          next_observation=double_transitions.next_observation[0],
          extras={"next_action": double_transitions.action[1]},
      )
  
file_path: acme/examples/offline/run_crr_jax.py
imports: from absl import app
  from absl import flags
  import acme
  from acme import specs
  from acme.agents.jax import actor_core
  from acme.agents.jax import actors
  from acme.agents.jax import crr
  from acme.datasets import tfds
  from acme.examples.offline import helpers
  from acme.jax import variable_utils
  from acme.types import Transition
  from acme.utils import loggers
  import haiku
  import jax
  import optax
  import rlds
label_desc: Use image node_id 1 for calling a global function with example usage: _add_next_action_extras(double_transitions) and returns: Transition