Fix bug in SimpleHeuristicsPlayer.choose_move

hsahovic · Jun 17, 2023 · 2fbe8f4 · 2fbe8f4
1 parent f975aa1
commit 2fbe8f4
Showing 1 changed file with 1 addition 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...