amend

facebookresearch · Nov 27, 2023 · 935e49e · 935e49e
1 parent 48395da
commit 935e49e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_pettingzoo.py b/test/test_pettingzoo.py
@@ -86,7 +86,7 @@ def test_all_tasks(
         experiment.run()
 
     @pytest.mark.parametrize("algo_config", [IppoConfig, QmixConfig, IsacConfig])
-    @pytest.mark.parametrize("task", [PettingZooTask.MULTIWALKER])
+    @pytest.mark.parametrize("task", [PettingZooTask.SIMPLE_TAG])
     def test_gnn(
         self,
         algo_config: AlgorithmConfig,