Default value of None for _tasks.pop to avoid KeyError

BQSKit · Oct 29, 2024 · 94a0ef5 · 94a0ef5
1 parent b7917dc
commit 94a0ef5
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: