Skip to content

Commit 3103f26

Browse files
committed
fix queue error
1 parent 8a60a34 commit 3103f26

2 files changed

Lines changed: 3 additions & 2 deletions

File tree

EchoShotX-ai-private

echoshot_ai_server/services/worker_pool.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
import multiprocessing as mp
22
from multiprocessing import Process, Queue
3+
from queue import Empty
34
from typing import List
45
import signal
56
import time
@@ -107,7 +108,7 @@ def _worker_loop(self, worker_id: int) -> None:
107108
self.sqs_client.delete_message(job.receipt_handle)
108109
logger.error(f"Job {job.job_id} exceeded max retries")
109110

110-
except Queue.Empty:
111+
except Empty:
111112
continue
112113
except Exception as e:
113114
logger.error(f"Worker {worker_id} error: {e}", exc_info=True)

0 commit comments

Comments
 (0)