also fixed issue for connect 4

rfali · Apr 2, 2021 · 5e328e8 · 5e328e8
1 parent aad38be
commit 5e328e8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pettingzoo/classic/connect_four/connect_four.py b/pettingzoo/classic/connect_four/connect_four.py
@@ -52,7 +52,7 @@ def __init__(self):
     #        [1, 1, 2, 1, 0, 1, 0]], dtype=int8)
     def observe(self, agent):
         board_vals = np.array(self.board).reshape(6, 7)
-        cur_player = self.possible_agents.index(self.agent_selection)
+        cur_player = self.possible_agents.index(agent)
         opp_player = (cur_player + 1) % 2
 
         cur_p_board = np.equal(board_vals, cur_player + 1)