(self, wrapper, num_dequeue_records=1)
| 16 | |
| 17 | class _QueueReader(dataio.Reader): |
| 18 | def __init__(self, wrapper, num_dequeue_records=1): |
| 19 | assert wrapper.schema is not None, ( |
| 20 | 'Queue needs a schema in order to be read from.') |
| 21 | dataio.Reader.__init__(self, wrapper.schema()) |
| 22 | self._wrapper = wrapper |
| 23 | self._num_dequeue_records = num_dequeue_records |
| 24 | |
| 25 | def setup_ex(self, init_net, exit_net): |
| 26 | exit_net.CloseBlobsQueue([self._wrapper.queue()], 0) |