diff --git a/game/pretense/pretenseluagenerator.py b/game/pretense/pretenseluagenerator.py index 7c5dc964..b10f6ab6 100644 --- a/game/pretense/pretenseluagenerator.py +++ b/game/pretense/pretenseluagenerator.py @@ -56,7 +56,7 @@ class PretenseLuaGenerator(LuaGenerator): self.mission.triggerrules.triggers.remove(t) self.mission.triggerrules.triggers.append(t) - def generate_pretense_land_upgrade_supply(self, cp_name: str, cp_side: int): + def generate_pretense_land_upgrade_supply(self, cp_name: str, cp_side: int) -> str: lua_string_zones = "" cp_name_trimmed = "".join([i for i in cp_name.lower() if i.isalnum()]) cp_side_str = "blue" if cp_side == PRETENSE_BLUE_SIDE else "red" diff --git a/game/pretense/pretensetgogenerator.py b/game/pretense/pretensetgogenerator.py index bcb9a0c2..74f99a6e 100644 --- a/game/pretense/pretensetgogenerator.py +++ b/game/pretense/pretensetgogenerator.py @@ -450,8 +450,8 @@ class PretenseTgoGenerator(TgoGenerator): else: other_coalition = coalition country = self.m.country(other_coalition.faction.country.name) + new_ground_object: TheaterGroundObject for ground_object in cp.ground_objects: - generator: GroundObjectGenerator if isinstance(ground_object, BuildingGroundObject): new_ground_object = BuildingGroundObject( name=ground_object.name,