diff --git a/deploy/gpu_worker/gpu_worker.py b/deploy/gpu_worker/gpu_worker.py index 8745597b7..72bb73aed 100644 --- a/deploy/gpu_worker/gpu_worker.py +++ b/deploy/gpu_worker/gpu_worker.py @@ -315,20 +315,20 @@ class TaskHeartbeat(threading.Thread): super().__init__(daemon=True, name=f"hb-{task_id[:8]}") self.task_id = task_id self.interval = max(5.0, interval) - self._stop = threading.Event() + self._stop_event = threading.Event() def run(self) -> None: # 先立即发一次,再按间隔循环(首次心跳失败不影响主流程) - while not self._stop.is_set(): + while not self._stop_event.is_set(): try: if _register(self.task_id): logger.debug("任务 %s 心跳已发送", self.task_id) except Exception as exc: # noqa: BLE001 logger.warning("任务 %s 心跳异常(忽略): %s", self.task_id, exc) - self._stop.wait(self.interval) + self._stop_event.wait(self.interval) def stop(self) -> None: - self._stop.set() + self._stop_event.set() def _handle_task(task: dict) -> None: