diff --git a/model.py b/model.py index 65bd010..9598cc4 100644 --- a/model.py +++ b/model.py @@ -130,7 +130,7 @@ def get_value(string): if string == "AIR": return UnitTarget.AIR if string == "BOTH": - return SpellType.BOTH + return UnitTarget.BOTH return None diff --git a/world.py b/world.py index a69efd4..0dba562 100644 --- a/world.py +++ b/world.py @@ -262,7 +262,7 @@ def _handle_turn_message(self, msg): self.player.hand = [self._get_base_unit_by_id(hand_type_id) for hand_type_id in msg["hand"]] self._handle_turn_kings(msg["kings"]) self._handle_turn_units(msg["units"]) - self._handle_turn_units(msg=msg["diedUnits"], is_dead_unit=True) + #self._handle_turn_units(msg=msg["diedUnits"], is_dead_unit=True) self._handle_turn_cast_spells(msg["castSpells"]) self.turn_updates = TurnUpdates(received_spell=msg["receivedSpell"],