Add new MuJoCo bindings (#2762)

This commit is contained in:
Rodrigo de Lazcano
2022-05-24 08:47:51 -04:00
committed by GitHub
parent 8f9b62f6a6
commit 3e006f3ea5
43 changed files with 3329 additions and 1523 deletions

View File

@@ -15,19 +15,25 @@ extras = {
"accept-rom-license": ["autorom[accept-rom-license]~=0.4.2"],
"box2d": ["box2d-py==2.3.5", "pygame==2.1.0"],
"classic_control": ["pygame==2.1.0"],
"mujoco": ["mujoco_py>=1.50, <2.0"],
"mujoco_py": ["mujoco_py<2.2,>=2.1"],
"mujoco": ["mujoco==2.1.5", "imageio>=2.14.1"],
"toy_text": ["pygame==2.1.0", "scipy>=1.4.1"],
"other": ["lz4>=3.1.0", "opencv-python>=3.0", "matplotlib>=3.0"],
}
# Meta dependency groups.
nomujoco_blacklist = {"mujoco", "accept-rom-license", "atari"}
nomujoco_blacklist = {"mujoco_py", "mujoco", "accept-rom-license", "atari"}
nomujoco_groups = set(extras.keys()) - nomujoco_blacklist
extras["nomujoco"] = list(
itertools.chain.from_iterable(map(lambda group: extras[group], nomujoco_groups))
)
noatari_blacklist = {"accept-rom-license", "atari"}
noatari_groups = set(extras.keys()) - noatari_blacklist
extras["noatari"] = list(
itertools.chain.from_iterable(map(lambda group: extras[group], noatari_groups))
)
all_blacklist = {"accept-rom-license"}
all_groups = set(extras.keys()) - all_blacklist