Skip to content

Commit e726fe5

Browse files
simplify
1 parent 5d0becc commit e726fe5

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

src/poke_env/environment/abstract_battle.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -218,15 +218,12 @@ def get_pokemon(
218218
if identifier[3] != " ":
219219
identifier = identifier[:2] + identifier[3:]
220220

221-
if identifier in self._team:
222-
return self._team[identifier]
223-
elif identifier in self._opponent_team:
224-
return self._opponent_team[identifier]
225-
226221
player_role = identifier[:2]
227222
name = identifier[3:].strip()
228223
is_mine = player_role == self._player_role
229224
team = self._team if is_mine or force_self_team else self._opponent_team
225+
if identifier in team:
226+
return team[identifier]
230227

231228
# if the pokemon has a nickname, this ensures we recognize it
232229
split_details = [
@@ -246,7 +243,6 @@ def get_pokemon(
246243
else:
247244
self._opponent_team = dict(items)
248245
team = self._team if is_mine or force_self_team else self._opponent_team
249-
250246
if identifier in team:
251247
return team[identifier]
252248

0 commit comments

Comments
 (0)