def registerModelVersionRelation(self, request, context):
model_version = transform_model_version_relation_meta(
request.model_version_relation)
response = self.store.register_model_version_relation(
version=model_version.version,
model_id=model_version.model_id,
workflow_execution_id=model_version.workflow_execution_id)
return _wrap_meta_response(
MetaToProto.model_version_relation_meta_to_proto(response))
def getModelVersionRelationByVersion(self, request, context):
model_version_meta = self.store.get_model_version_relation_by_version(request.name, request.model_id)
return _wrap_meta_response(MetaToProto.model_version_relation_meta_to_proto(model_version_meta))
