Terminate python sub-process

pull/907/head
yanghaitao 5 years ago
parent e433f1e70a
commit 1e98fa35b8

@ -1431,6 +1431,10 @@ class MapDataset(DatasetOp):
iter_specific_operations.append(op)
self.operations = iter_specific_operations
def __del__(self):
if hasattr(self, 'process_pool') and self.process_pool is not None:
self.process_pool.terminate()
class FilterDataset(DatasetOp):
"""
@ -2463,6 +2467,9 @@ class _GeneratorWorker(multiprocessing.Process):
"""
return self.res_queue.get(timeout=5)
def __del__(self):
self.terminate()
class GeneratorDataset(SourceDataset):
"""

Loading…
Cancel
Save