make initial thread pool optional
This commit is contained in:
parent
6426cff741
commit
c3b0c5e3e3
|
@ -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))
|
||||||
|
|
Loading…
Reference in New Issue