Move unit purchase off find_unittype.

This commit is contained in:
Dan Albert
2021-06-07 18:54:21 -07:00
parent ea15421308
commit 27b5f24a0f
4 changed files with 23 additions and 28 deletions

View File

@@ -1332,7 +1332,7 @@ def upgrade_to_supercarrier(unit, name: str):
def find_unittype(for_task: Type[MainTask], country_name: str) -> List[Type[UnitType]]:
return [x for x in UNIT_BY_TASK[for_task] if x in FACTIONS[country_name].units]
return [x for x in UNIT_BY_TASK[for_task] if x in FACTIONS[country_name].all_units]
MANPADS: List[Type[VehicleType]] = [