From 3e13bb8dc85ae9164a00748f970ded5cf6d9c574 Mon Sep 17 00:00:00 2001 From: Peter Zhokhov Date: Fri, 5 Apr 2019 15:13:59 -0700 Subject: [PATCH] short-circuit framestack wrapper with size 1 --- baselines/common/retro_wrappers.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/baselines/common/retro_wrappers.py b/baselines/common/retro_wrappers.py index 2c42926..badbbdd 100644 --- 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