bugfix: use reentrant-safe SimpleQueue instead of queue.Queue

alephdata · Dec 19, 2024 · d9cbeb8 · d9cbeb8
1 parent fa82787
commit d9cbeb8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/servicelayer/taskqueue.py b/servicelayer/taskqueue.py
@@ -9,7 +9,7 @@
 import sys
 from abc import ABC, abstractmethod
 import functools
-from queue import Queue, Empty
+from queue import SimpleQueue, Empty
 import platform
 from collections import defaultdict
 from threading import Thread
@@ -450,7 +450,7 @@ def __init__(
         self.num_threads = num_threads
         self.queues = ensure_list(queues)
         self.version = version
-        self.local_queue = Queue()
+        self.local_queue = SimpleQueue()
         self.prefetch_count_mapping = prefetch_count_mapping
         if settings.SENTRY_DSN:
             import sentry_sdk