class_docstr: Creates (s,a,r,s',a') transitions.
class_docstr_tok: ["Creates","(","s",",","a",",","r",",","s","'",",","a","'",")","transitions","."]
code: def __init__(
      self,
      stack_size,
      update_horizon,
      gamma,
      next_action=False,
      return_to_go=False,
  ):
      self.stack_size = stack_size
      self.update_horizon = update_horizon
      self.total_frames = stack_size + update_horizon
      self.cumulative_discount = tf.pow(gamma, range(update_horizon))
      self.next_action = next_action
      self.return_to_go = return_to_go
  
file_path: dopamine/dopamine/labs/offline_rl/rlu_tfds/tfds_atari_utils.py
imports: import rlds
  import tensorflow
  import tensorflow_datasets
label_desc: Use image node_id 1 to create a new BatchToTransition object from inherited base classes: object with example: obj = BatchToTransition(stack_size, update_horizon, gamma, next_action, return_to_go)