Formatting + type annotations

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:

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