From 60f837d0b98b14326fdce4ac0e494e895d6bf370 Mon Sep 17 00:00:00 2001 From: Khopa Date: Mon, 21 Dec 2020 13:34:56 +0100 Subject: [PATCH] Fixed : AI wouldn't buy artillery units --- game/procurement.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/game/procurement.py b/game/procurement.py index 6852c534..7c33dd40 100644 --- a/game/procurement.py +++ b/game/procurement.py @@ -75,14 +75,14 @@ class ProcurementAi: def random_affordable_ground_unit( self, budget: int) -> Optional[Type[VehicleType]]: - affordable_units = [u for u in self.faction.frontline_units if + affordable_units = [u for u in self.faction.frontline_units + self.faction.artillery_units if db.PRICES[u] <= budget] if not affordable_units: return None return random.choice(affordable_units) def reinforce_front_line(self, budget: int) -> int: - if not self.faction.frontline_units: + if not self.faction.frontline_units and not self.faction.artillery_units: return budget while budget > 0: