def create_vncatari_env(env_id, client_id, remotes, **_):
env = gym.make(env_id)
env = Vision(env)
env = Logger(env)
env = BlockingReset(env)
env = GymCoreAction(env)
env = AtariRescale42x42(env)
#env = NormalizedEnv(env)
env = EpisodeID(env)
env = DiagnosticsInfo(env)
if "multi_obs_settings" in _.keys():
_multi = [env] + list(_["multi_obs_settings"])
env = [env, SplitScreen(*_multi)]
#env = Unvectorize(env)
logger.info('Connecting to remotes: %s', remotes)
fps = env.metadata['video.frames_per_second']
if not "no_config" in _.keys():
env.configure(remotes=remotes,
start_timeout=15 * 60,
fps=fps,
client_id=client_id)
return env
def create_vncatari_env(env_id, client_id, remotes, **_):
"""Create an Atari (rescaled) environment by passing environment id.
Parameters
----------
env_id : str
environment id to be registered in Gym
client_id : str
Client ID
remotes : str
BLANK
kwargs : dict
BLANK
"""
env = gym.make(env_id)
env = Vision(env)
env = Logger(env)
env = BlockingReset(env)
env = GymCoreAction(env)
env = AtariRescale42x42(env)
env = EpisodeID(env)
env = DiagnosticsInfo(env)
env = Unvectorize(env)
logger.info('Connecting to remotes: %s', remotes)
fps = env.metadata['video.frames_per_second']
env.configure(remotes=remotes,
start_timeout=15 * 60,
fps=fps,
client_id=client_id)
return env
def create_vncatari_env(env_id, client_id, remotes, **_):
env = gym.make(env_id)
env = Vision(env)
env = BlockingReset(env)
env = GymCoreAction(env)
env = AtariRescale(env)
env = Unvectorize(env)
print('Connecting to remotes: {}'.format(remotes))
env.configure(remotes=remotes,
start_timeout=15 * 60,
fps=env.metadata['video.frames_per_second'],
client_id=client_id)
return env
def create_vncatari_env(env_id, client_id, remotes, **_):
env = gym.make(env_id)
env = Vision(env)
env = Logger(env)
env = BlockingReset(env)
env = GymCoreAction(env)
env = AtariRescale(env)
env = EpisodeID(env)
env = DiagnosticsInfo(env)
env = Unvectorize(env)
logger.info('Connecting to remotes: %s', remotes)
fps = env.metadata['video.frames_per_second']
env.configure(remotes=remotes, start_timeout=15 * 60, fps=fps, client_id=client_id)
return env
