fix bug in PrefetchDataZMQ

da062119 · Yuxin Wu · 68e8d9eb · da062119
Commit da062119 authored Dec 09, 2017 by Yuxin Wu
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

tensorpack/dataflow/prefetch.py tensorpack/dataflow/prefetch.py +2 -1

No files found.
--- a/tensorpack/dataflow/prefetch.py
+++ b/tensorpack/dataflow/prefetch.py
@@ -247,8 +247,9 @@ class PrefetchDataZMQ(_MultiProcessZMQDataFlow):
            nr_proc (int): number of processes to use.
            hwm (int): the zmq "high-water mark" (queue size) for both sender and receiver.
        """
-        super(PrefetchDataZMQ, self).__init__(ds)
+        super(PrefetchDataZMQ, self).__init__()

+        self.ds = ds
        self.nr_proc = nr_proc
        self._hwm = hwm