Jofthomas commited on
Commit
297ff84
·
1 Parent(s): 5625887

remove batlle typing too

Browse files
Files changed (1) hide show
  1. agents.py +2 -2
agents.py CHANGED
@@ -144,7 +144,7 @@ class LLMAgentBase(Player):
144
  f"Opponent Side Conditions: {battle.opponent_side_conditions}"
145
  return state_str.strip()
146
 
147
- def _find_move_by_name(self, battle, move_name: str) -> Optional[Move]:
148
  normalized_name = normalize_name(move_name)
149
  # Prioritize exact ID match
150
  for move in battle.available_moves:
@@ -157,7 +157,7 @@ class LLMAgentBase(Player):
157
  return move
158
  return None
159
 
160
- def _find_pokemon_by_name(self, battle, pokemon_name: str) -> Optional[Pokemon]:
161
  normalized_name = normalize_name(pokemon_name)
162
  for pkmn in battle.available_switches:
163
  # Normalize the species name for comparison
 
144
  f"Opponent Side Conditions: {battle.opponent_side_conditions}"
145
  return state_str.strip()
146
 
147
+ def _find_move_by_name(self, battle, move_name: str):
148
  normalized_name = normalize_name(move_name)
149
  # Prioritize exact ID match
150
  for move in battle.available_moves:
 
157
  return move
158
  return None
159
 
160
+ def _find_pokemon_by_name(self, battle, pokemon_name: str):
161
  normalized_name = normalize_name(pokemon_name)
162
  for pkmn in battle.available_switches:
163
  # Normalize the species name for comparison