def make_env(nenv):
"""Create a training environment."""
return make_atari_env("Pong", nenv)
def env_fn(nenv):
return make_atari_env('Pong', nenv, frame_stack=1)
def make_env(nenv):
"""Create a training environment."""
return VecEpisodeLogger(VecObsNormWrapper(make_atari_env("Pong",
nenv)))