Merge pull request #300 from BQSKit/default_task_pop

Fix `KeyError` for already popped tasks
BQSKit · Nov 1, 2024 · 14eab3c · 14eab3c
2 parents b7917dc + 94a0ef5
commit 14eab3c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bqskit/runtime/worker.py b/bqskit/runtime/worker.py
@@ -524,7 +524,7 @@ def _process_task_completion(self, task: RuntimeTask, result: Any) -> None:
             self._conn.send((RuntimeMessage.RESULT, packaged_result))
 
         # Remove task
-        self._tasks.pop(task.return_address)
+        self._tasks.pop(task.return_address, None)
 
         # Cancel any open tasks
         for mailbox_id in self._active_task.owned_mailboxes: