def METHOD_NAME(self) -> np.ndarray:
    self.game_over = False
    timestep = self._env.METHOD_NAME()
    self._last_observation = timestep.observation
    return np.reshape(timestep.observation,
                      (timestep.observation.shape[1], ))