fix #33 (#34)

2020-04-21 15:36:08 +08:00 · 2020-04-21 15:36:08 +08:00 · 205698dd66
commit 205698dd66
parent 4fd826761c
1 changed files with 3 additions and 2 deletions
--- a/tianshou/env/atari.py
+++ b/tianshou/env/atari.py
@ -95,8 +95,9 @@ class preprocessing(object):
                axis=-1)
        if self.count >= self.max_episode_steps:
            terminal = True
-        self.terminal = terminal
-        return observation, total_reward, is_terminal, info
+        else:
+            terminal = False
+        return observation, total_reward, (terminal or is_terminal), info

    def _grayscale_obs(self, output):
        self.env.ale.getScreenGrayscale(output)