From 5d5730335626f5b06c574cdfaf68fccb28962d7f Mon Sep 17 00:00:00 2001 From: Austin Mroz Date: Tue, 25 Jun 2024 20:22:50 -0500 Subject: [PATCH] Fix mistaken index on priority update --- workflowcheckpointing.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/workflowcheckpointing.py b/workflowcheckpointing.py index c112d50..ea7c0aa 100644 --- a/workflowcheckpointing.py +++ b/workflowcheckpointing.py @@ -136,7 +136,7 @@ class FetchQueue: with self.lock: if item in self.consumed: #TODO: Also update in queue - self.consumed[item][0] = min(self.consumed[item][1], priority) + self.consumed[item][1] = min(self.consumed[item][1], priority) return self.consumed[item][0] for i in range(len(self.queue)): if self.queue[i][2] == item: @@ -180,8 +180,8 @@ class FetchLoop: with self.queue.lock: if url in self.queue.consumed: self.queue.consumed.pop(url) - async def enqueue(self, url, priority=0): - return await self.queue.enqueue_checked(url, priority) + def enqueue(self, url, priority=0): + return self.queue.enqueue_checked(url, priority) async def fetch(self, priority, url, future): chunk_size = 2**25 #32MB headers = {}