We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 2f3aeb3 commit ecade76Copy full SHA for ecade76
job_pool/job_pool.py
@@ -59,7 +59,6 @@ class AbnormalWorkerTerminationError(Exception):
59
60
61
class JobPool:
62
- @threadpool_limits.wrap(limits=1, user_api='blas')
63
def __init__(
64
self,
65
processes: int = 1,
@@ -121,8 +120,7 @@ def checkPool(self, printProgressEvery: int = -1):
121
120
):
122
self.checkForTerminatedProcess(res)
123
outputs.append(res.get())
124
- self.pool.close()
125
- self.pool.join()
+ self.stopPool()
126
return outputs
127
except (KeyboardInterrupt, SystemExit) as e:
128
logger.error(f"Caught {e.__class__.__name__}, terminating workers")
0 commit comments