def get_ram_env(env, seed):
set_global_seeds(seed)
env.seed(seed)
expt_dir = '/tmp/gym-results'
env = wrappers.Monitor(env, expt_dir, force=True)
env = wrap_deepmind_ram(env)
return env
def get_env_by_id(env_id, seed, skip_f=4, clip_max=False):
env = gym.make(env_id)
set_global_seeds(seed)
env.seed(seed)
expt_dir = 'tmp/gym-results'
env = wrap_deepmind_nomaxclip(env, skip_f)
return env
def get_env(task, seed, monitor=True):
env = gym.make(task)
set_global_seeds(seed)
env.seed(seed)
expt_dir = "tmp/gym-results"
if monitor:
env = wrappers.Monitor(env, expt_dir, force=True)
env = wrap_deepmind(env)
return env
def get_env(env_id, seed):
env = gym.make(env_id)
set_global_seeds(seed)
env.seed(seed)
expt_dir = 'tmp/gym-results'
env = wrappers.Monitor(env, expt_dir, force=True)
env = wrap_deepmind(env)
return env
def get_env(task, seed):
env_id = task.env_id
env = gym.make(env_id)
set_global_seeds(seed)
env.seed(seed)
expt_dir = 'results/pong'
env = wrappers.Monitor(env, expt_dir, force=True)
env = wrap_deepmind(env)
return env
def get_env(seed):
# env_id = task.env_id
env = gym.make('MsPacman-v0')
set_global_seeds(seed)
env.seed(seed)
expt_dir = 'tmp/gym-results'
env = wrappers.Monitor(env, expt_dir, force=True)
env = wrap_deepmind(env)
return env
def get_ram_env(env, seed):
set_global_seeds(seed)
env.seed(seed)
expt_dir = '/tmp/gym-results'
env = wrappers.Monitor(
env,
expt_dir,
force=True,
video_callable=lambda episode_id: episode_id % 100 == 0)
env = wrap_deepmind_ram(env)
return env
def get_env(env, seed):
# env_id = task.env_id
# env = gym.make(env_id)
set_global_seeds(seed)
env.seed(seed)
expt_dir = 'tmp/gym-results'
env = wrappers.Monitor(env, expt_dir, force=True)
print(env.observation_space)
print(env.spec.id)
env = wrap_deepmind(env)
print("get_env")
return env
def get_env(task, seed):
env_id = task.env_id
env = gym.make(env_id)
set_global_seeds(seed)
env.seed(seed)
expt_dir = 'tmp/gym-results'
env = wrappers.Monitor(
env,
expt_dir,
force=True,
video_callable=lambda episode_id: episode_id % 100 == 0)
env = wrap_deepmind(env)
return env
