(self, handler, schema=None, num_dequeue_records=1)
| 57 | |
| 58 | class QueueWrapper(dataio.Pipe): |
| 59 | def __init__(self, handler, schema=None, num_dequeue_records=1): |
| 60 | dataio.Pipe.__init__(self, schema, TaskGroup.LOCAL_SETUP) |
| 61 | self._queue = handler |
| 62 | self._num_dequeue_records = num_dequeue_records |
| 63 | |
| 64 | def reader(self): |
| 65 | return _QueueReader( |