From 935e49ec063fb8d2f02d8ffe2c40700588e7cb1d Mon Sep 17 00:00:00 2001
From: Matteo Bettini <mb2389@cl.cam.ac.uk>
Date: Mon, 27 Nov 2023 15:47:34 +0000
Subject: [PATCH] amend

---
 test/test_pettingzoo.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/test_pettingzoo.py b/test/test_pettingzoo.py
index 7a04fa5f..7d86a37d 100644
--- 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,