Fix to SafeQueue for mac

luigifreda · Dec 20, 2024 · 912a166 · 912a166
1 parent d28d9ac
commit 912a166
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/utilities/utils_data.py b/utilities/utils_data.py
@@ -144,6 +144,14 @@ def get(self, block=True, timeout=None):
         with self._lock:
             self._size.value -= 1
         return item
+
+    def get_nowait(self):
+        """Remove and return an item from the queue without blocking."""
+        item = self.queue.get_nowait()
+        with self._lock:
+            if item:
+                self._size.value -= 1
+        return item
 
     def qsize(self):
         """Return the current size of the queue."""