def METHOD_NAME(self, action: int) -> _GymTimestep:
    timestep = self._env.METHOD_NAME(action)
    self._last_observation = timestep.observation
    reward = timestep.reward or 0.
    if timestep.last():
        self.game_over = True
    return np.reshape(
        timestep.observation,
        (timestep.observation.shape[1], )), reward, timestep.last(), {}