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)