Fix bug in SimpleHeuristicsPlayer.choose_move

hsahovic · Jun 17, 2023 · 186baf3 · 186baf3
1 parent f975aa1
commit 186baf3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/poke_env/player/baselines.py b/src/poke_env/player/baselines.py
@@ -121,7 +121,7 @@ def choose_move(self, battle):
                 [m for m in battle.team.values() if m.fainted is False]
             )
             n_opp_remaining_mons = 6 - len(
-                [m for m in battle.team.values() if m.fainted is True]
+                [m for m in battle.opponent_team.values() if m.fainted is True]
             )
 
             # Entry hazard...

diff --git a/unit_tests/player/test_baselines.py b/unit_tests/player/test_baselines.py
@@ -180,6 +180,7 @@ def test_simple_heuristics_player():
             "available_moves",
             "available_switches",
             "team",
+            "opponent_team",
             "can_dynamax",
             "side_conditions",
             "opponent_side_conditions",
@@ -191,6 +192,7 @@ def test_simple_heuristics_player():
         [],
         [Pokemon(species="togekiss", gen=8)],
         {},
+        {},
         True,
         set(),
         set(),