def get_execution_types_by_id(
self,
type_ids: Iterable[int]) -> List[metadata_store_pb2.ExecutionType]:
"""Gets execution types by ID.
Args:
type_ids: a sequence of execution type IDs.
Returns:
A list of execution types.
Args:
type_ids: ids to look for.
Raises:
InternalError: if query execution fails.
"""
request = metadata_store_service_pb2.GetExecutionTypesByIDRequest()
response = metadata_store_service_pb2.GetExecutionTypesByIDResponse()
for x in type_ids:
request.type_ids.append(x)
self._call('GetExecutionTypesByID', request, response)
result = []
for x in response.execution_types:
result.append(x)
return result
def get_execution_types_by_id(
self,
type_ids: Sequence[int]) -> List[metadata_store_pb2.ExecutionType]:
"""Gets types by ID.
TODO(b/122657258): implement in gRPC
Args:
type_ids: a sequence of artifact type IDs.
Returns:
A list of execution types.
Args:
type_ids: ids to look for.
Raises:
InternalError: if query execution fails.
"""
request = metadata_store_service_pb2.GetExecutionTypesByIDRequest()
response = metadata_store_service_pb2.GetExecutionTypesByIDResponse()
for x in type_ids:
request.type_ids.append(x)
self._swig_call(metadata_store_serialized.GetExecutionTypesByID,
request, response)
result = []
for x in response.execution_types:
result.append(x)
return result
