call remove_done_callback in finally section (#1688)

This commit is contained in:
jimmylai 2017-05-22 22:32:46 -07:00 committed by Łukasz Langa
parent 002665a9da
commit 21b3e04c13
1 changed files with 2 additions and 1 deletions

View File

@ -458,7 +458,8 @@ def run_until_complete(self, future):
# local task. # local task.
future.exception() future.exception()
raise raise
future.remove_done_callback(_run_until_complete_cb) finally:
future.remove_done_callback(_run_until_complete_cb)
if not future.done(): if not future.done():
raise RuntimeError('Event loop stopped before Future completed.') raise RuntimeError('Event loop stopped before Future completed.')