From 5b37698d360712199455272b7ca599a616978045 Mon Sep 17 00:00:00 2001 From: Khopa Date: Tue, 3 Aug 2021 00:28:22 +0200 Subject: [PATCH] Fixed mypy errors --- game/game.py | 3 ++- game/helipad.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/game/game.py b/game/game.py index 2305523a..e931335a 100644 --- a/game/game.py +++ b/game/game.py @@ -8,6 +8,7 @@ from typing import Any, List, Type, Union, cast from dcs.action import Coalition from dcs.countries import Switzerland, UnitedNationsPeacekeepers, USAFAggressors +from dcs.country import Country from dcs.mapping import Point from dcs.task import CAP, CAS, PinpointStrike from dcs.vehicles import AirDefence @@ -189,7 +190,7 @@ class Game: ) @property - def neutral_country(self): + def neutral_country(self) -> Type[Country]: """Return the best fitting country that can be used as neutral faction in the generated mission""" countries_in_use = [self.red.country_name, self.blue.country_name] if UnitedNationsPeacekeepers not in countries_in_use: diff --git a/game/helipad.py b/game/helipad.py index 0f113d00..4146556b 100644 --- a/game/helipad.py +++ b/game/helipad.py @@ -8,7 +8,7 @@ from game.utils import Heading class Helipad(PointWithHeading): - def __init__(self): + def __init__(self) -> None: super(Helipad, self).__init__() self.heading = Heading.from_degrees(0) self.occupied = False