def complete_all_trajectories(self):
    """Essentially same as reset, but we don't have observations."""
    for index in range(self.batch_size):
      trajectory = self._trajectories[index]
      # TODO(pkozakowski): This assertion breaks something in SimPLe trajectory
      # collection code - we're probably doing something wrong there. Commenting
      # out the assertion as a temporary measure.
      # assert trajectory.is_active
      if trajectory.is_active:
        self._complete_trajectory(trajectory, index)
Essentially same as reset, but we don't have observations.
tensor2tensor/envs/trajectory.py