From 978e3e4de2d9334de3fbb86faa9caa041944cbfb Mon Sep 17 00:00:00 2001 From: Raffson Date: Thu, 18 May 2023 16:29:20 +0200 Subject: [PATCH] Formatting + type annotations --- game/missiongenerator/missiongenerator.py | 3 ++- game/persistency.py | 4 ++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/game/missiongenerator/missiongenerator.py b/game/missiongenerator/missiongenerator.py index e9f545b1..1510a10c 100644 --- a/game/missiongenerator/missiongenerator.py +++ b/game/missiongenerator/missiongenerator.py @@ -135,7 +135,8 @@ class MissionGenerator: belligerents = {self.p_country.id, self.e_country.id} for country_id in country_dict.keys(): if country_id not in belligerents: - self.mission.coalition["neutrals"].add_country(country_dict[country_id]()) + c = country_dict[country_id]() + self.mission.coalition["neutrals"].add_country(c) def add_airfields_to_unit_map(self) -> None: for control_point in self.game.theater.controlpoints: diff --git a/game/persistency.py b/game/persistency.py index d594e898..b95b0f41 100644 --- a/game/persistency.py +++ b/game/persistency.py @@ -4,7 +4,7 @@ import logging import pickle import shutil from pathlib import Path -from typing import Optional, TYPE_CHECKING +from typing import Optional, TYPE_CHECKING, Any from game.profiling import logged_duration @@ -17,7 +17,7 @@ _dcs_saved_game_folder: Optional[str] = None # fmt: off class MigrationUnpickler(pickle.Unpickler): """Custom unpickler to migrate campaign save-files for when components have been moved""" - def find_class(self, module, name): + def find_class(self, module: Any, name: str) -> Any: if name == "NightMissions": from game.weather.conditions import NightMissions return NightMissions