Fix typing issue

hsahovic · Apr 6, 2024 · c352e35 · c352e35
1 parent f994ebd
commit c352e35
Showing 1 changed file with 18 additions and 8 deletions.
diff --git a/src/poke_env/environment/abstract_battle.py b/src/poke_env/environment/abstract_battle.py
@@ -695,14 +695,24 @@ def parse_message(self, split_message: List[str]):
                 return
             if username == self._player_username:
                 self._player_role = player
-            return self._players.append(
-                {
-                    "username": username,
-                    "player": player,
-                    "avatar": avatar,
-                    "rating": rating,
-                }
-            )
+            if rating is not None:
+                return self._players.append(
+                    {
+                        "username": username,
+                        "player": player,
+                        "avatar": avatar,
+                        "rating": rating,
+                    }
+                )
+            else:
+                return self._players.append(
+                    {
+                        "username": username,
+                        "player": player,
+                        "avatar": avatar,
+                    }
+                )
+
         elif split_message[1] == "poke":
             player, details = split_message[2:4]
             self._register_teampreview_pokemon(player, details)