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

separate start_process from prefetch

parent 1095c8b8
...@@ -125,12 +125,14 @@ class PrefetchDataZMQ(ProxyDataFlow): ...@@ -125,12 +125,14 @@ class PrefetchDataZMQ(ProxyDataFlow):
self.procs = [PrefetchProcessZMQ(self.ds, self.pipename) self.procs = [PrefetchProcessZMQ(self.ds, self.pipename)
for _ in range(self.nr_proc)] for _ in range(self.nr_proc)]
start_proc_mask_signal(self.procs) self.start_processes()
# __del__ not guranteed to get called at exit # __del__ not guranteed to get called at exit
import atexit import atexit
atexit.register(lambda x: x.__del__(), self) atexit.register(lambda x: x.__del__(), self)
def start_processes(self):
start_proc_mask_signal(self.procs)
def get_data(self): def get_data(self):
for k in itertools.count(): for k in itertools.count():
if self._size > 0 and k >= self._size: 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