apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

make initial thread pool optional

This commit is contained in:
Sean Sube 2023-02-04 10:07:58 -06:00
parent 6426cff741
commit c3b0c5e3e3
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

@ -76,7 +76,7 @@ class DevicePoolExecutor:
jobs: List[Job] = None jobs: List[Job] = None
pool: Union[ProcessPoolExecutor, ThreadPoolExecutor] = None pool: Union[ProcessPoolExecutor, ThreadPoolExecutor] = None
def __init__(self, devices: List[str], pool: Optional[Union[ProcessPoolExecutor, ThreadPoolExecutor]]): def __init__(self, devices: List[str], pool: Optional[Union[ProcessPoolExecutor, ThreadPoolExecutor]] = None):
self.devices = devices self.devices = devices
self.jobs = [] self.jobs = []
self.pool = pool or ThreadPoolExecutor(len(devices)) self.pool = pool or ThreadPoolExecutor(len(devices))