diff --git a/priority_queue.py b/priority_queue.py index 9a0087b..51d0000 100644 --- a/priority_queue.py +++ b/priority_queue.py @@ -12,10 +12,10 @@ def pop(self): self._swap(0, len(self.heap) - 1) item = self.heap.pop() self._sift_down(0) - return item + return item[1] elif len(self.heap) == 1: item = self.heap.pop() - return item + return item[1] else: return None