[RLlib; docs] Fix RLlib main index page (code in multi-agent tab wron…

…g). (#49037)
ray-project · Dec 4, 2024 · 19f9d88 · 19f9d88
1 parent 1b13782
commit 19f9d88
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/doc/source/rllib/index.rst b/doc/source/rllib/index.rst
@@ -193,8 +193,14 @@ click the dropdowns below:
     |                                                                                           |
     | .. code-block:: python                                                                    |
     |                                                                                           |
-    |     config.environment("CartPole-v1")  # Classic Control                                  |
-    |     config.environment("ale_py:ALE/Pong-v5")  # Atari                                     |
+    |     from ray.rllib.examples.envs.classes.multi_agent import MultiAgentCartPole            |
+    |     from ray import tune                                                                  |
+    |     tune.register_env("env", lambda cfg: MultiAgentCartPole(cfg))                         |
+    |     config.environment("env", env_config={"num_agents": 2})                               |
+    |     config.multi_agent(                                                                   |
+    |         policies={"p0", "p1"},                                                            |
+    |         policy_mapping_fn=lambda aid, *a, **kw: f"p{aid}",                                |
+    |     )                                                                                     |
     +-------------------------------------------------------------------------------------------+