code: def get_q_values(model, states, actions, do_rollout, rng):
      results = model(
          states, actions=actions, do_rollout=do_rollout, key=rng
      )[0]
      return results.q_values, results.latent
  
imports: import collections
  import copy
  import functools
  import time
  from absl import logging
  from dopamine.jax import losses
  from dopamine.jax.agents.dqn import dqn_agent
  from dopamine.jax.agents.rainbow import rainbow_agent
  from dopamine.labs.atari_100k import atari_100k_rainbow_agent
  from dopamine.labs.atari_100k.replay_memory import subsequence_replay_buffer
  from dopamine.labs.atari_100k.spr_networks import SPRNetwork
  import gin
  import jax
  import jax.numpy
  import optax
  import tensorflow
label_desc: Use image node_id 2 for calling a global function with example usage: get_q_values(model, states, actions, do_rollout, rng) and returns: results,  results