Python exception_type_from_error_code Examples

Example #1

File: queue_runner_impl.py Project: AutumnQYN/tensorflow

  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,)

Example #2

File: queue_runner_impl.py Project: wwjiang007/tensorflow

    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, )

Example #3

File: core.py Project: TheVinhLuong102/tensorflow

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)

Example #4

File: metadata_store.py Project: rauanarchive/ml-metadata

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)

Example #5

File: metadata_store.py Project: jlemire-u3d/ml-metadata

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)

Example #6

File: rpc_ops.py Project: zhuochenKIDD/tensorflow

 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())

Example #7

File: core.py Project: Harryi0/tinyML

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)

Example #8

File: core.py Project: Dr4KK/tensorflow

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)

Example #9

File: metadata_store.py Project: bhlarson/ml-metadata

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)