Fix in 'unit_task' method

This commit is contained in:
Khopa 2019-07-03 21:06:03 +02:00
parent a17a9399ef
commit a30d4a4514

View File

@ -940,8 +940,10 @@ def unit_task(unit: UnitType) -> Task:
for task, units in UNIT_BY_TASK.items():
if unit in units:
return task
elif unit in SAM_CONVERT:
return SAM_CONVERT[unit]
if unit in SAM_CONVERT:
return unit_task(SAM_CONVERT[unit])
assert False