From 2cd468448eaa023294d8b7e9a8a7d1be0a6ebf54 Mon Sep 17 00:00:00 2001 From: Philip Paquette Date: Thu, 6 Jun 2019 12:11:32 -0400 Subject: Power - Clearing dislodged units on clear_units() --- diplomacy/engine/power.py | 3 +++ 1 file changed, 3 insertions(+) (limited to 'diplomacy/engine/power.py') diff --git a/diplomacy/engine/power.py b/diplomacy/engine/power.py index 570359b..1fe282b 100644 --- a/diplomacy/engine/power.py +++ b/diplomacy/engine/power.py @@ -275,7 +275,10 @@ class Power(Jsonable): """ Removes all units from the map """ for unit in self.units: self.game.update_hash(self.name, unit_type=unit[0], loc=unit[2:]) + for unit in self.retreats: + self.game.update_hash(self.name, unit_type=unit[0], loc=unit[2:], is_dislodged=True) self.units = [] + self.retreats = {} self.influence = [] self.game.clear_cache() -- cgit v1.2.3