def METHOD_NAME(self, action: int) -> RLStepResult:
    assert isinstance(action, int)
    minigrid_obs, reward, done, info = self.env.step(action=action)
    self._last_action = action
    self._was_successful = done and reward > 0
    return RLStepResult(
        observation=self.get_observations(minigrid_output_obs=minigrid_obs),
        reward=reward,
        done=self.is_done(),
        info=info,
    )