fixB

Taomercy · Feb 2, 2019 · 35380cb · 35380cb
1 parent 073d551
commit 35380cb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/sc2/bot_ai.py b/sc2/bot_ai.py
@@ -593,6 +593,8 @@ def _prepare_step(self, state):
         self.supply_cap: Union[float, int] = state.common.food_cap
         self.supply_workers: Union[float, int] = state.common.food_workers  # Doesn't include workers in production
         self.supply_army: Union[float, int] = state.common.food_army
+        self.supply_workers: Union[float, int] = state.common.food_workers  # Doesn't include workers in production
+        self.supply_army: Union[float, int] = state.common.food_army
         self.supply_left: Union[float, int] = self.supply_cap - self.supply_used
         # reset cached values
         self.cached_known_enemy_structures = None

diff --git a/sc2/game_state.py b/sc2/game_state.py
@@ -108,6 +108,7 @@ def __init__(self, response_observation, game_data):
         # https://github.com/Blizzard/s2client-proto/blob/51662231c0965eba47d5183ed0a6336d5ae6b640/s2clientprotocol/sc2api.proto#L575
         self.observation = response_observation.observation
         self.observation_raw = self.observation.raw_data
+        self.alerts = self.observation.alerts
         self.player_result = response_observation.player_result
         self.chat = response_observation.chat
         self.common: Common = Common(self.observation.player_common)