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