apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

only return device index

This commit is contained in:
Sean Sube 2023-02-04 16:40:18 -06:00
parent bf3f227d66
commit de6e3b4e27
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/device_pool.py
View File

@ -146,7 +146,7 @@ class DevicePoolExecutor:
queued = job_counts.most_common() queued = job_counts.most_common()
logger.debug('jobs queued by device: %s', queued) logger.debug('jobs queued by device: %s', queued)
return queued[-1] return queued[-1][0]
def prune(self): def prune(self):
self.jobs[:] = [job for job in self.jobs if job.future.done()] self.jobs[:] = [job for job in self.jobs if job.future.done()]