diff --git a/distdeepq/static.py b/distdeepq/static.py index 6d9d3d7..397ab0b 100644 --- a/distdeepq/static.py +++ b/distdeepq/static.py @@ -43,7 +43,7 @@ def make_env(game_name): from baselines.common.atari_wrappers import wrap_deepmind, make_atari env = make_atari(game_name + "NoFrameskip-v4") monitored_env = SimpleMonitor(env) - env = wrap_deepmind(monitored_env) + env = wrap_deepmind(monitored_env, frame_stack=True, scale=True) return env, monitored_env