def _init_from_proto(self, queue_runner_def, import_scope=None): """Create a QueueRunner from `QueueRunnerDef`. Args: queue_runner_def: Optional `QueueRunnerDef` protocol buffer. import_scope: Optional `string`. Name scope to add. """ assert isinstance(queue_runner_def, queue_runner_pb2.QueueRunnerDef) g = ops.get_default_graph() self._queue = g.as_graph_element( ops.prepend_name_scope(queue_runner_def.queue_name, import_scope)) self._enqueue_ops = [g.as_graph_element( ops.prepend_name_scope(op, import_scope)) for op in queue_runner_def.enqueue_op_name] self._close_op = g.as_graph_element(ops.prepend_name_scope( queue_runner_def.close_op_name, import_scope)) self._cancel_op = g.as_graph_element(ops.prepend_name_scope( queue_runner_def.cancel_op_name, import_scope)) self._queue_closed_exception_types = tuple( errors.exception_type_from_error_code(code) for code in queue_runner_def.queue_closed_exception_types) # Legacy support for old QueueRunnerDefs created before this field # was added. if not self._queue_closed_exception_types: self._queue_closed_exception_types = (errors.OutOfRangeError,)
def _init_from_proto(self, queue_runner_def, import_scope=None): """Create a QueueRunner from `QueueRunnerDef`. Args: queue_runner_def: Optional `QueueRunnerDef` protocol buffer. import_scope: Optional `string`. Name scope to add. """ assert isinstance(queue_runner_def, queue_runner_pb2.QueueRunnerDef) g = ops.get_default_graph() self._queue = g.as_graph_element( ops.prepend_name_scope(queue_runner_def.queue_name, import_scope)) self._enqueue_ops = [ g.as_graph_element(ops.prepend_name_scope(op, import_scope)) for op in queue_runner_def.enqueue_op_name ] self._close_op = g.as_graph_element( ops.prepend_name_scope(queue_runner_def.close_op_name, import_scope)) self._cancel_op = g.as_graph_element( ops.prepend_name_scope(queue_runner_def.cancel_op_name, import_scope)) self._queue_closed_exception_types = tuple( errors.exception_type_from_error_code(code) for code in queue_runner_def.queue_closed_exception_types) # Legacy support for old QueueRunnerDefs created before this field # was added. if not self._queue_closed_exception_types: self._queue_closed_exception_types = (errors.OutOfRangeError, )
def _status_to_exception(status): try: error_class = errors.exception_type_from_error_code(status.code) return error_class(None, None, status.message, status.payloads) except KeyError: return errors.UnknownError(None, None, status.message, status.code, status.payloads)
def _make_exception(message, error_code): try: exc_type = errors.exception_type_from_error_code(error_code) logging.log(logging.ERROR, 'mlmd client %s: %s', exc_type.__name__, message) return exc_type(None, None, message) except KeyError: return errors.UnknownError(None, None, message, error_code)
def _make_exception(msg, error_code): try: exc_type = errors.exception_type_from_error_code(error_code) # log internal backend engine errors only. if error_code == errors.INTERNAL: logging.log(logging.WARNING, 'mlmd client %s: %s', exc_type.__name__, msg) return exc_type(None, None, msg) except KeyError: return errors.UnknownError(None, None, msg, error_code)
def call_blocking_wrapper(*args, timeout_in_ms=0): status_or, deleter = gen_rpc_ops.rpc_call( client_handle, args=validate_and_get_flat_inputs(*args), method_name=method_name, timeout_in_ms=timeout_in_ms) status_or = StatusOrResult(status_or, deleter, output_specs) if status_or.is_ok(): return status_or.get_value() else: error_code, error_msg = status_or.get_error() raise errors.exception_type_from_error_code( error_code.numpy())(None, None, error_msg.numpy())
def _status_to_exception(code, message): try: error_class = errors.exception_type_from_error_code(code) return error_class(None, None, message) except KeyError: return errors.UnknownError(None, None, message, code)
def _make_exception(message, error_code): try: exc_type = errors.exception_type_from_error_code(error_code) return exc_type(None, None, message) except KeyError: return errors.UnknownError(None, None, message, error_code)