hongshaorou/Tianshou
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix #33 (#34)

Browse Source
This commit is contained in:
Minghao Zhang 2020-04-21 15:36:08 +08:00 committed by GitHub
parent 4fd826761c
commit 205698dd66
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tianshou/env/atari.py vendored
View File

@ -95,8 +95,9 @@ class preprocessing(object):
axis=-1) axis=-1)
if self.count >= self.max_episode_steps: if self.count >= self.max_episode_steps:
terminal = True terminal = True
self.terminal = terminal else:
return observation, total_reward, is_terminal, info terminal = False
return observation, total_reward, (terminal or is_terminal), info
def _grayscale_obs(self, output): def _grayscale_obs(self, output):
self.env.ale.getScreenGrayscale(output) self.env.ale.getScreenGrayscale(output)