[Feature] Update list of supported libraries (#594)

pytorch · Oct 21, 2022 · 4cbc521 · 4cbc521
1 parent 6933f70
commit 4cbc521
Showing 1 changed file with 20 additions and 21 deletions.
diff --git a/torchrl/envs/utils.py b/torchrl/envs/utils.py
@@ -134,30 +134,29 @@ def _check_dmlab():
 SUPPORTED_LIBRARIES = {
     "gym": _check_gym(),  # OpenAI
     "gym[atari]": _check_gym_atari(),  #
-    "vizdoom": None,  # 1.2k, https://github.com/mwydmuch/ViZDoom
-    "ml-agents": None,
-    # 11.5k, unity, https://github.com/Unity-Technologies/ml-agents
-    "pysc2": None,  # 7.3k, DM, https://github.com/deepmind/pysc2
-    "deepmind_lab": _check_dmlab(),
-    # 6.5k DM, https://github.com/deepmind/lab, https://github.com/deepmind/lab/tree/master/python/pip_package
-    "serpent.ai": None,  # 6k, https://github.com/SerpentAI/SerpentAI
-    "gfootball": None,  # 2.8k G, https://github.com/google-research/football
     "dm_control": _check_dmcontrol(),
-    # 2.3k DM, https://github.com/deepmind/dm_control
     "habitat": None,
-    # 1.2k FB, https://github.com/facebookresearch/habitat-sim
-    "meta-world": None,  # 500, https://github.com/rlworkgroup/metaworld
-    "minerl": None,  # 300, https://github.com/minerllabs/minerl
-    "multi-agent-emergence-environments": None,
-    # 1.2k, OpenAI, https://github.com/openai/multi-agent-emergence-environments
-    "openspiel": None,  # 2.8k, DM, https://github.com/deepmind/open_spiel
-    "procgen": None,  # 500, OpenAI, https://github.com/openai/procgen
-    "pybullet": None,  # 641, https://github.com/benelot/pybullet-gym
-    "realworld_rl_suite": None,
-    # 250, G, https://github.com/google-research/realworldrl_suite
-    "rlcard": None,  # 1.4k, https://github.com/datamllab/rlcard
-    "screeps": None,  # 2.3k https://github.com/screeps/screeps
     "gym-super-mario-bros": _check_mario(),
+    # "vizdoom": None,  # gym based, https://github.com/mwydmuch/ViZDoom
+    # "openspiel": None,  # DM, https://github.com/deepmind/open_spiel
+    # "pysc2": None,  # DM, https://github.com/deepmind/pysc2
+    # "deepmind_lab": _check_dmlab(),
+    # DM, https://github.com/deepmind/lab, https://github.com/deepmind/lab/tree/master/python/pip_package
+    # "serpent.ai": None,  # https://github.com/SerpentAI/SerpentAI
+    # "gfootball": None,  # 2.8k G, https://github.com/google-research/football
+    # DM, https://github.com/deepmind/dm_control
+    # FB, https://github.com/facebookresearch/habitat-sim
+    # "meta-world": None,  # https://github.com/rlworkgroup/metaworld
+    # "minerl": None,  # https://github.com/minerllabs/minerl
+    # "multi-agent-emergence-environments": None,
+    # OpenAI, https://github.com/openai/multi-agent-emergence-environments
+    # "procgen": None,  # OpenAI, https://github.com/openai/procgen
+    # "pybullet": None,  # https://github.com/benelot/pybullet-gym
+    # "realworld_rl_suite": None,
+    # G, https://github.com/google-research/realworldrl_suite
+    # "rlcard": None,  # https://github.com/datamllab/rlcard
+    # "screeps": None,  # https://github.com/screeps/screeps
+    # "ml-agents": None,
 }
 
 EXPLORATION_MODE = None