[RLlib] Refine MultiAgentEpisode class and add test cases. (ray-pro…

…ject#40799)
ujjawal-khare-27 · Nov 29, 2023 · 278c60c · 278c60c
1 parent 1ef373e
commit 278c60c
Show file tree

Hide file tree

Showing 5 changed files with 3,497 additions and 223 deletions.
diff --git a/rllib/BUILD b/rllib/BUILD
@@ -772,6 +772,13 @@ py_test(
     srcs = ["env/tests/test_multi_agent_env.py"]
 )
 
+py_test(
+    name = "env/tests/test_multi_agent_episode",
+    tags = ["team:rllib", "env"],
+    size = "medium",
+    srcs = ["env/tests/test_multi_agent_episode.py"]
+)
+
 sh_test(
     name = "env/tests/test_remote_inference_cartpole",
     tags = ["team:rllib", "env", "exclusive"],