def _data_block_handler(self, msg): logging.debug('DataBlock: recv "%s" at %d', msg.block_id, msg.count) assert self._count == msg.count if not msg.block_id: block = None elif self._trainer_master is not None: block = self._trainer_master.request_data_block(msg.block_id) return False else: block = DataBlockInfo(msg.block_id, None) self._count += 1 self._block_queue.put(block) return True
def _data_block_handler(self, msg): logging.debug('DataBlock: recv "%s" at %d', msg.block_id, msg.count) assert self._count == msg.count if not msg.block_id: block = None elif self._trainer_master is not None: block = self._trainer_master.request_data_block(msg.block_id) if block is None: raise ValueError("Block %s not found" % msg.block_id) else: block = DataBlockInfo(msg.block_id, None) self._count += 1 self._block_queue.put(block)