29573a73ddf5cb1b3cd89301c84f0057c04cc084,mushroom_rl/environments/gym_env.py,Gym,step,#Gym#Any#,68

Before Change

    def step(self, action):
        action = self._convert_action(action)

        return self.env.step(action)

    def render(self, mode="human"):
        self.env.render(mode=mode)

After Change

        action = self._convert_action(action)
        obs, reward, absorbing, info = self.env.step(action)

        return np.atleast_1d(obs), reward, absorbing, info

    def render(self, mode="human"):
        self.env.render(mode=mode)

Instances