From ac7657defd33080e3161cd2b5761d7e86d1b5ebc Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 4 Feb 2023 10:50:41 -0600 Subject: [PATCH] add debug logs to progress --- api/onnx_web/device_pool.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/api/onnx_web/device_pool.py b/api/onnx_web/device_pool.py index abbd129b..ff243acc 100644 --- a/api/onnx_web/device_pool.py +++ b/api/onnx_web/device_pool.py @@ -33,7 +33,9 @@ class JobContext: if device_index < 0: raise Exception('job has not been assigned to a device') else: - return self.devices[device_index] + device = self.devices[device_index] + logger.debug('job %s assigned to device %s', self.key, device) + return device def get_progress(self) -> int: return self.progress.value @@ -54,6 +56,7 @@ class JobContext: def set_progress(self, progress: int) -> None: with self.progress.get_lock(): self.progress.value = progress + logger.debug('setting progress for job %s to %s', self.key, progress) class Job: