def _get_experiment():
request_message = _get_request_message(GetExperiment())
response_message = GetExperiment.Response()
experiment = _get_tracking_store().get_experiment(request_message.experiment_id).to_proto()
response_message.experiment.MergeFrom(experiment)
response = Response(mimetype='application/json')
response.set_data(message_to_json(response_message))
return response
def _get_experiment():
request_message = _get_request_message(GetExperiment())
response_message = GetExperiment.Response()
response_message.experiment.MergeFrom(_get_store().get_experiment(request_message.experiment_id)
.to_proto())
run_info_entities = _get_store().list_run_infos(request_message.experiment_id)
response_message.runs.extend([r.to_proto() for r in run_info_entities])
response = Response(mimetype='application/json')
response.set_data(_message_to_json(response_message))
return response
def _get_experiment():
request_message = _get_request_message(GetExperiment(), from_get=True)
response_message = GetExperiment.Response()
response_message.experiment.MergeFrom(
store.get_experiment(request_message.experiment_id).to_proto())
run_info_entities = store.list_run_infos(request_message.experiment_id)
response_message.runs.extend([r.to_proto() for r in run_info_entities])
response = Response(mimetype='application/json')
response.set_data(
MessageToJson(response_message, preserving_proto_field_name=True))
return response
def get_experiment(self, experiment_id):
"""
Fetches the experiment from the backend store.
:param experiment_id: Integer id for the experiment
:return: A single Experiment object if it exists, otherwise raises an Exception.
"""
req_body = message_to_json(GetExperiment(experiment_id=experiment_id))
response_proto = self._call_endpoint(GetExperiment, req_body)
return Experiment.from_proto(response_proto.experiment)
def get_experiment(self, experiment_id):
"""
Fetch the experiment from the backend store.
:param experiment_id: String id for the experiment
:return: A single :py:class:`mlflow.entities.Experiment` object if it exists,
otherwise raises an Exception.
"""
req_body = message_to_json(GetExperiment(experiment_id=str(experiment_id)))
response_proto = self._call_endpoint(GetExperiment, req_body)
return Experiment.from_proto(response_proto.experiment)
