Fixed mypy errors

This commit is contained in:
Khopa 2021-08-03 00:28:22 +02:00
parent 483640b0c6
commit 5b37698d36
2 changed files with 3 additions and 2 deletions

View File

@ -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:

View File

@ -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