def from_args(cls, args, seed, **kwargs):
env = gym_super_mario_bros.make(args.env)
env = JoypadSpace(env, [["right"], ["right", "A"]])
env = SkipFrame(env, skip=args.skip_rate)
env._max_episode_steps = args.max_episode_length
env.seed(seed)
return cls(
env, args.frame_stack, args.skip_rate, args.max_episode_length
)
def get_marioenv(world=1, stage=1, version=0, movement = RIGHT_ONLY, max_episode_steps=4500):
mario_env = JoypadSpace(SuperMario_Env(world, stage, version), movement)
mario_env._max_episode_steps = max_episode_steps * 4
mario_env = StickyActionEnv(mario_env)
mario_env = MaxAndSkipEnv(mario_env, skip=4)
mario_env = DummyMontezumaInfoWrapper(mario_env)
mario_env = AddRandomStateToInfo(mario_env)
mario_env = wrap_deepmind(mario_env, frame_stack=True)
return mario_env
def _thunk():
mario_env = JoypadSpace(SuperMario_Env(world, stage, version),
movement)
if wrap_atari:
mario_env._max_episode_steps = max_episode_steps * 4
mario_env = StickyActionEnv(mario_env)
mario_env = MaxAndSkipEnv(mario_env, skip=4)
mario_env = DummyMontezumaInfoWrapper(mario_env)
mario_env = AddRandomStateToInfo(mario_env)
# mario_env.seed(seed + rank)
mario_env = Monitor(
mario_env,
logger.get_dir()
and os.path.join(logger.get_dir(), str(rank)),
allow_early_resets=True)
if wrap_atari:
mario_env = wrap_deepmind(mario_env)
mario_env = BlocksWrapper(mario_env)
mario_env.seed(seed)
return mario_env
