short-circuit framestack wrapper with size 1

2019-04-05 15:13:59 -07:00
parent c79b3373bf
commit 3e13bb8dc8
1 changed files with 2 additions and 1 deletions
--- a/baselines/common/retro_wrappers.py
+++ b/baselines/common/retro_wrappers.py
@@ -215,7 +215,8 @@ def wrap_deepmind_retro(env, scale=True, frame_stack=4):
    """
    env = WarpFrame(env)
    env = ClipRewardEnv(env)
-    env = FrameStack(env, frame_stack)
+    if frame_stack > 1:
+        env = FrameStack(env, frame_stack)
    if scale:
        env = ScaledFloatFrame(env)
    return env