apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

add debug logs to progress

This commit is contained in:
Sean Sube 2023-02-04 10:50:41 -06:00
parent 900a95eb61
commit ac7657defd
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
api/onnx_web/device_pool.py
View File

@ -33,7 +33,9 @@ class JobContext:
if device_index < 0: if device_index < 0:
raise Exception('job has not been assigned to a device') raise Exception('job has not been assigned to a device')
else: 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: def get_progress(self) -> int:
return self.progress.value return self.progress.value
@ -54,6 +56,7 @@ class JobContext:
def set_progress(self, progress: int) -> None: def set_progress(self, progress: int) -> None:
with self.progress.get_lock(): with self.progress.get_lock():
self.progress.value = progress self.progress.value = progress
logger.debug('setting progress for job %s to %s', self.key, progress)
class Job: class Job: