class_docstr: A compact implementation of agents for Atari 100k.
code: def restore_train_state(self):
      (
          self.state,
          self._last_observation,
          self._observation,
      ) = self.training_state
  
imports: import copy
  import functools
  from absl import logging
  from dopamine.jax import networks
  from dopamine.jax.agents.full_rainbow import full_rainbow_agent
  import gin
  import jax
  import jax.numpy
  import numpy
  import tensorflow
label_desc: Use image node_id 10 for calling the Atari100kRainbowAgent obj's underlying member method code with example usage: obj.restore_train_state() without return types