lambda evaluate, render, n_levels, multiagent, shared, maddpg:
BipedalObstacles(render=render),
},
# ======================================================================= #
# Point navigation environments. #
# ======================================================================= #
"Point2DEnv": {
"meta_ac_space":
lambda relative_goals, multiagent: Box(
np.ones(2) * -4, np.ones(2) * 4, dtype=np.float32),
"state_indices":
lambda multiagent: [0, 1],
"env":
lambda evaluate, render, n_levels, multiagent, shared, maddpg:
Point2DEnv(images_in_obs=False),
},
"Point2DImageEnv": {
"meta_ac_space":
lambda relative_goals, multiagent: Box(
np.ones(2) * -4, np.ones(2) * 4, dtype=np.float32),
"state_indices":
lambda multiagent: [3072, 3073],
"env":
lambda evaluate, render, n_levels, multiagent, shared, maddpg:
Point2DEnv(images_in_obs=True),
},
}
def _get_ring_env_attributes(scale):
[i + 1024 for i in [0, 4, 5, 6, 7, 32, 33, 34, 50, 51, 52]],
"env":
lambda evaluate, render, multiagent, shared, maddpg: BipedalObstacles(
render=render),
},
# ======================================================================= #
# Point navigation environments. #
# ======================================================================= #
"Point2DEnv": {
"meta_ac_space":
lambda relative_goals: Box(
np.ones(2) * -4, np.ones(2) * 4, dtype=np.float32),
"state_indices": [0, 1],
"env":
lambda evaluate, render, multiagent, shared, maddpg: Point2DEnv(
images_in_obs=False),
},
"Point2DImageEnv": {
"meta_ac_space":
lambda relative_goals: Box(
np.ones(2) * -4, np.ones(2) * 4, dtype=np.float32),
# "state_indices": [3072, 3073], # FIXME: for RGB
"state_indices": [1024, 1025],
"env":
lambda evaluate, render, multiagent, shared, maddpg: Point2DEnv(
images_in_obs=True),
},
}
def get_meta_ac_space(ob_space, relative_goals, env_name, use_fingerprints,