Commit 5a31fc32 authored by Yuxin Wu's avatar Yuxin Wu

separate start_process from prefetch

parent 1095c8b8
......@@ -125,12 +125,14 @@ class PrefetchDataZMQ(ProxyDataFlow):
self.procs = [PrefetchProcessZMQ(self.ds, self.pipename)
for _ in range(self.nr_proc)]
start_proc_mask_signal(self.procs)
self.start_processes()
# __del__ not guranteed to get called at exit
import atexit
atexit.register(lambda x: x.__del__(), self)
def start_processes(self):
start_proc_mask_signal(self.procs)
def get_data(self):
for k in itertools.count():
if self._size > 0 and k >= self._size:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment