def get_execution_type(self, type_name):
"""Gets an execution type, or None if it does not exist."""
request = metadata_store_service_pb2.GetExecutionTypeRequest()
request.type_name = type_name
response = metadata_store_service_pb2.GetExecutionTypeResponse()
self._swig_call(metadata_store_serialized.GetExecutionType, request,
response)
return response.execution_type
def get_execution_type(
self, type_name: Text) -> metadata_store_pb2.ExecutionType:
"""Gets an execution type by name.
Args:
type_name: the type with that name.
Returns:
The type with name type_name.
Raises:
tensorflow.errors.NotFoundError: if no type exists
tensorflow.errors.InternalError: if query execution fails
"""
request = metadata_store_service_pb2.GetExecutionTypeRequest()
request.type_name = type_name
response = metadata_store_service_pb2.GetExecutionTypeResponse()
self._call('GetExecutionType', request, response)
return response.execution_type
