tools/asyncio_wait_or_cancel: wait for cancellation

2015-05-21 16:41:27 +08:00 · 2015-05-21 16:41:27 +08:00 · a670b9f660
commit a670b9f660
parent 6c35d066fc
1 changed files with 1 additions and 0 deletions
--- a/artiq/tools.py
+++ b/artiq/tools.py
@ -109,6 +109,7 @@ def asyncio_wait_or_cancel(fs, **kwargs):
        raise
    for f in p:
        f.cancel()
+        yield from asyncio.wait([f])
    return fs