Refresh task queue first before get task from queue (#139)

aiidateam · Dec 2, 2024 · f3cdb40 · f3cdb40
1 parent 7059d24
commit f3cdb40
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/kiwipy/rmq/tasks.py b/src/kiwipy/rmq/tasks.py
@@ -119,6 +119,9 @@ async def next_task(self,
         raises:
             kiwipy.exceptions.QueueEmpty: When the queue has no tasks within the timeout
         """
+        # relinquish so that if there is requeue coroutines, they are run first and task queue get updated
+        await asyncio.sleep(0)
+
         try:
             message = await self._task_queue.get(no_ack=no_ack, fail=fail, timeout=timeout)
         except aio_pika.exceptions.QueueEmpty as exc: