diff --git a/game/theater/theatergroundobject.py b/game/theater/theatergroundobject.py index acca4417..2a9f0870 100644 --- a/game/theater/theatergroundobject.py +++ b/game/theater/theatergroundobject.py @@ -163,7 +163,6 @@ class TheaterGroundObject(MissionTarget, SidcDescribable, ABC): else: yield from [ FlightType.STRIKE, - FlightType.BAI, FlightType.REFUELING, ] yield from super().mission_types(for_player) @@ -602,6 +601,13 @@ class VehicleGroupGroundObject(TheaterGroundObject): def should_head_to_conflict(self) -> bool: return True + def mission_types(self, for_player: bool) -> Iterator[FlightType]: + from game.ato import FlightType + + if not self.is_friendly(for_player): + yield FlightType.BAI + yield from super().mission_types(for_player) + class EwrGroundObject(IadsGroundObject): def __init__(