def get_registered_model_details(self, registered_model):
"""
:param registered_model: :py:class:`mlflow.entities.model_registry.RegisteredModel` object.
:return: A single :py:class:`mlflow.entities.model_registry.RegisteredModelDetailed` object.
"""
req_body = message_to_json(GetRegisteredModelDetails(
registered_model=registered_model.to_proto()))
response_proto = self._call_endpoint(GetRegisteredModelDetails, req_body)
return RegisteredModelDetailed.from_proto(response_proto.registered_model_detailed)
def list_registered_models(self):
"""
List of all registered models.
:return: List of :py:class:`mlflow.entities.model_registry.RegisteredModel` objects.
"""
req_body = message_to_json(ListRegisteredModels())
response_proto = self._call_endpoint(ListRegisteredModels, req_body)
return [RegisteredModelDetailed.from_proto(registered_model_detailed)
for registered_model_detailed in response_proto.registered_models_detailed]