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

fix atari wrapper to be deterministic (#467)

Browse Source
This commit is contained in:
Jiayi Weng 2021-10-19 10:26:11 -04:00 committed by GitHub
parent b9eedc516e
commit 098d466467
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/atari/atari_wrapper.py
View File

@ -24,7 +24,7 @@ class NoopResetEnv(gym.Wrapper):
def reset(self): def reset(self):
self.env.reset() self.env.reset()
noops = np.random.randint(1, self.noop_max + 1) noops = self.unwrapped.np_random.randint(1, self.noop_max + 1)
for _ in range(noops): for _ in range(noops):
obs, _, done, _ = self.env.step(self.noop_action) obs, _, done, _ = self.env.step(self.noop_action)
if done: if done: