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) iter_specific_operations.append(op)
self.operations = iter_specific_operations 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): class FilterDataset(DatasetOp):
""" """
@ -2463,6 +2467,9 @@ class _GeneratorWorker(multiprocessing.Process):
""" """
return self.res_queue.get(timeout=5) return self.res_queue.get(timeout=5)
def __del__(self):
self.terminate()
class GeneratorDataset(SourceDataset): class GeneratorDataset(SourceDataset):
""" """

Loading…
Cancel
Save