code: def setUp(self):
      super(FullRainbowAgentTest, self).setUp()
      self._num_actions = 4
      self._num_atoms = 5
      self._vmax = 7.0
      self.observation_shape = dqn_agent.NATURE_DQN_OBSERVATION_SHAPE
      self.observation_dtype = dqn_agent.NATURE_DQN_DTYPE
      self.stack_size = dqn_agent.NATURE_DQN_STACK_SIZE
      self.zero_state = onp.zeros(
          (1,) + self.observation_shape + (self.stack_size,)
      )
      gin.bind_parameter(
          "OutOfGraphPrioritizedReplayBuffer.replay_capacity", 100
      )
      gin.bind_parameter(
          "OutOfGraphPrioritizedReplayBuffer.batch_size", 2
      )
      gin.bind_parameter("JaxDQNAgent.min_replay_history", 32)
      gin.bind_parameter("JaxDQNAgent.epsilon_eval", 0.0)
      gin.bind_parameter("JaxDQNAgent.epsilon_decay_period", 90)
  
file_path: dopamine/tests/dopamine/jax/agents/full_rainbow/full_rainbow_agent_test.py
imports: from absl.testing import absltest
  from dopamine.discrete_domains import atari_lib
  from dopamine.jax.agents.dqn import dqn_agent
  from dopamine.jax.agents.full_rainbow import full_rainbow_agent
  from dopamine.utils import test_utils
  from flax import linen
  import gin
  import jax.numpy
  import numpy
label_desc: Use image node_id 1 for calling the FullRainbowAgentTest obj's underlying member method code with example usage: obj.setUp() without return types