Compare commits

...

1 Commits

Author SHA1 Message Date
Peter Zhokhov
3e13bb8dc8 short-circuit framestack wrapper with size 1 2019-04-05 15:13:59 -07:00

View File

@@ -215,7 +215,8 @@ def wrap_deepmind_retro(env, scale=True, frame_stack=4):
""" """
env = WarpFrame(env) env = WarpFrame(env)
env = ClipRewardEnv(env) env = ClipRewardEnv(env)
env = FrameStack(env, frame_stack) if frame_stack > 1:
env = FrameStack(env, frame_stack)
if scale: if scale:
env = ScaledFloatFrame(env) env = ScaledFloatFrame(env)
return env return env