提交 b7433598 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!372 use custom response instead of response

Merge pull request !372 from 治愈系潇洒哥/r0.5
...@@ -21,7 +21,6 @@ import json ...@@ -21,7 +21,6 @@ import json
import os import os
from flask import Blueprint from flask import Blueprint
from flask import Response
from flask import jsonify from flask import jsonify
from flask import request from flask import request
from marshmallow import ValidationError from marshmallow import ValidationError
...@@ -40,6 +39,7 @@ from mindinsight.profiler.common.validator.validate_path import \ ...@@ -40,6 +39,7 @@ from mindinsight.profiler.common.validator.validate_path import \
from mindinsight.profiler.common.validator.validate_path import validate_and_normalize_profiler_path from mindinsight.profiler.common.validator.validate_path import validate_and_normalize_profiler_path
from mindinsight.profiler.proposer.compose_proposer import ComposeProposal from mindinsight.profiler.proposer.compose_proposer import ComposeProposal
from mindinsight.utils.exceptions import ParamValueError from mindinsight.utils.exceptions import ParamValueError
from mindinsight.backend.application import CustomResponse
BLUEPRINT = Blueprint("profile", __name__, url_prefix=settings.URL_PATH_PREFIX+settings.API_PREFIX) BLUEPRINT = Blueprint("profile", __name__, url_prefix=settings.URL_PATH_PREFIX+settings.API_PREFIX)
...@@ -320,7 +320,7 @@ def get_profile_summary_proposal(): ...@@ -320,7 +320,7 @@ def get_profile_summary_proposal():
proposal_obj = ComposeProposal(profiler_dir_abs, device_id, proposal_type_list) proposal_obj = ComposeProposal(profiler_dir_abs, device_id, proposal_type_list)
proposal_info = proposal_obj.get_proposal(options) proposal_info = proposal_obj.get_proposal(options)
# Use json.dumps for orderly return # Use json.dumps for orderly return
return Response(json.dumps(proposal_info), mimetype='application/json') return CustomResponse(json.dumps(proposal_info), mimetype='application/json')
@BLUEPRINT.route("/profile/minddata-pipeline/op-queue", methods=["POST"]) @BLUEPRINT.route("/profile/minddata-pipeline/op-queue", methods=["POST"])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册