def insert(self, step, current_obs, state, action, action_log_prob, value_pred, reward, mask):
    self.observations[(step + 1)].copy_(current_obs)
    self.states[(step + 1)].copy_(state)
    self.actions[step].copy_(action)
    self.action_log_probs[step].copy_(action_log_prob)
    self.value_preds[step].copy_(value_pred)
    self.rewards[step].copy_(reward)
    self.masks[(step + 1)].copy_(mask)
def insert(self, step, current_obs, state, action, action_log_prob, value_pred, reward, mask):
    self.observations[(step + 1)].copy_(current_obs)
    self.states[(step + 1)].copy_(state)
    self.actions[step].copy_(action)
    self.action_log_probs[step].copy_(action_log_prob)
    self.value_preds[step].copy_(value_pred)
    self.rewards[step].copy_(reward)
    self.masks[(step + 1)].copy_(mask)