class_docstr: Simple Python accumulator for transitions.
class_docstr_tok: ["Simple","Python","accumulator","for","transitions","."]
code: def sample(self, batch_size):
      assert batch_size == 1
      return Transition(
          self._prev.observation,
          self._action,
          self._latest.reward,
          self._latest.discount,
          self._latest.observation,
      )
  
imports: import collections
  from absl import app
  from absl import flags
  from bsuite.environments import catch
  from bsuite.utils import wrappers
  import haiku
  from haiku import nets
  import jax
  import jax.numpy
  import optax
  import rlax
  from rlax.examples import experiment
label_desc: Use image node_id 3 for calling the TransitionAccumulator obj's underlying member method code with example usage: obj.sample(batch_size) and returns: Transition