fix: 🐛 load energy and hp

sigureling · May 3, 2024 · 95f4c84 · 95f4c84
1 parent 4fe2090
commit 95f4c84
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/CAPI/python/PyAPI/utils.py b/CAPI/python/PyAPI/utils.py
@@ -279,7 +279,11 @@ def Protobuf2THUAI7GameInfo(
         gameInfo = THUAI7.GameInfo()
         gameInfo.gameTime = allMsg.game_time
         gameInfo.redScore = allMsg.red_team_score
+        gameInfo.redEnergy = allMsg.red_team_energy
+        gameInfo.redHomeHp = allMsg.red_home_hp
         gameInfo.blueScore = allMsg.blue_team_score
+        gameInfo.blueEnergy = allMsg.blue_team_energy
+        gameInfo.blueHomeHp = allMsg.blue_home_hp
         return gameInfo