thu-ml · Trinkle23897 · Feb 15, 2022 · Sep 11, 2021 · Sep 23, 2021 · Sep 23, 2021
diff --git a/setup.py b/setup.py
@@ -55,6 +55,7 @@ def get_version() -> str:
         "torch>=1.4.0",
         "numba>=0.51.0",
         "h5py>=2.10.0",  # to match tensorflow's minimal requirements
+        "pettingzoo>=1.12,<=1.13",
     ],
     extras_require={
         "dev": [
@@ -74,6 +75,9 @@ def get_version() -> str:
             "pydocstyle",
             "doc8",
             "scipy",
+            "pillow",
+            "pygame>=2.1.0",  # pettingzoo test cases pistonball
+            "pymunk>=6.2.1",  # pettingzoo test cases pistonball
         ],
         "atari": ["atari_py", "opencv-python"],
         "mujoco": ["mujoco_py"],

diff --git a/test/multiagent/Gomoku.py b/test/multiagent/Gomoku.py
diff --git a/test/multiagent/tic_tac_toe_env.py b/test/multiagent/tic_tac_toe_env.py