e8ffab75132e3788d52b8a8f14dcd63fb7888905,tensorforce/contrib/openai_gym.py,OpenAIGym,reset,#OpenAIGym#,63

Before Change

    def reset(self):
        if isinstance(self.gym, gym.wrappers.Monitor):
            self.gym.stats_recorder.done = True
        return self.gym.reset()

    def execute(self, action):
        if self.visualize:
            self.gym.render()

After Change

    def reset(self):
        if isinstance(self.gym, gym.wrappers.Monitor):
            self.gym.stats_recorder.done = True
        state = self.gym.reset()
        return OpenAIGym.flatten_state(state=state)

    def execute(self, action):
        if self.visualize:
            self.gym.render()

Instances