def _search_runs():
request_message = _get_request_message(SearchRuns())
response_message = SearchRuns.Response()
run_view_type = ViewType.ACTIVE_ONLY
if request_message.HasField('run_view_type'):
run_view_type = ViewType.from_proto(request_message.run_view_type)
run_entities = _get_store().search_runs(request_message.experiment_ids,
SearchFilter(request_message),
run_view_type)
response_message.runs.extend([r.to_proto() for r in run_entities])
response = Response(mimetype='application/json')
response.set_data(message_to_json(response_message))
return response
def _search_runs():
request_message = _get_request_message(SearchRuns())
response_message = SearchRuns.Response()
run_view_type = ViewType.ACTIVE_ONLY
if request_message.HasField('run_view_type'):
run_view_type = ViewType.from_proto(request_message.run_view_type)
sf = SearchFilter(anded_expressions=request_message.anded_expressions,
filter_string=request_message.filter)
max_results = request_message.max_results
experiment_ids = request_message.experiment_ids
run_entities = _get_store().search_runs(experiment_ids, sf, run_view_type, max_results)
response_message.runs.extend([r.to_proto() for r in run_entities])
response = Response(mimetype='application/json')
response.set_data(message_to_json(response_message))
return response
def _search_runs():
request_message = _get_request_message(SearchRuns())
response_message = SearchRuns.Response()
run_view_type = ViewType.ACTIVE_ONLY
if request_message.HasField('run_view_type'):
run_view_type = ViewType.from_proto(request_message.run_view_type)
filter_string = request_message.filter
max_results = request_message.max_results
experiment_ids = request_message.experiment_ids
order_by = request_message.order_by
page_token = request_message.page_token
run_entities = _get_store().search_runs(experiment_ids, filter_string,
run_view_type, max_results,
order_by, page_token)
response_message.runs.extend([r.to_proto() for r in run_entities])
response = Response(mimetype='application/json')
response.set_data(message_to_json(response_message))
return response
def test_from_proto():
assert ViewType.from_proto(service_pb2.ACTIVE_ONLY) == ViewType.ACTIVE_ONLY
assert ViewType.from_proto(
service_pb2.DELETED_ONLY) == ViewType.DELETED_ONLY
assert ViewType.from_proto(service_pb2.ALL) == ViewType.ALL
