def _make_env(self, name):
import ppaquette_gym_super_mario
from ppaquette_gym_super_mario import wrappers
name_split = name.split('-')
if name_split[0] != 'meta':
prefix, world, level = name_split[:3]
author, prefix = prefix.split('/')
suffix = '-'.join(name_split[3:])
self._env_name = '/'.join([author, '-'.join(['meta', prefix, suffix])])
self._env_level = (int(world) - 1) * 4 + int(level) - 1
else:
self._env_name = name
self._env_level = None
env = gym.make(self._env_name)
# modewrapper = wrappers.SetPlayingMode('algo')
return GymNintendoWrapper(env)
评论列表
文章目录