fix pooler so finisher is optional

This commit is contained in:
Connor Olding 2020-09-01 08:22:05 +02:00
parent 3b31919f75
commit 9a3598b599

View file

@ -64,6 +64,7 @@ def make_pooler(pool_size, finisher=None):
desired_size = 0 if finish else pool_size - 1 desired_size = 0 if finish else pool_size - 1
while len(pending) > desired_size: while len(pending) > desired_size:
done, pending = await wait(pending, return_when=FIRST_COMPLETED) done, pending = await wait(pending, return_when=FIRST_COMPLETED)
if finisher is not None:
finisher(done, pending) finisher(done, pending)
return pooler return pooler