Skip to content

Commit 7d0ded1

Browse files
fix erroneous training numbers
1 parent cf1431e commit 7d0ded1

1 file changed

Lines changed: 2 additions & 4 deletions

File tree

autotraining.lua

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -207,8 +207,7 @@ function addTraining(unit)
207207
for _, squad in ipairs(getTrainingSquads()) do
208208
for i=1,9,1 do
209209
if ( squad.positions[i].occupant == -1 ) then
210-
dfhack.military.addToSquad(unit.id,squad.id,i)
211-
return true
210+
return dfhack.military.addToSquad(unit.id,squad.id,i)
212211
end
213212
end
214213
end
@@ -220,8 +219,7 @@ function removeTraining(unit)
220219
for _, squad in ipairs(getTrainingSquads()) do
221220
for i=1,9,1 do
222221
if ( unit.hist_figure_id == squad.positions[i].occupant ) then
223-
dfhack.military.removeFromSquad(unit.id)
224-
return true
222+
return dfhack.military.removeFromSquad(unit.id)
225223
end
226224
end
227225
end

0 commit comments

Comments
 (0)