diff --git a/mindinsight/backend/profiler/profile_api.py b/mindinsight/backend/profiler/profile_api.py index 887cf4f2c8f6f1db95329fc05e0b4a4eef786e5f..cda97a4b8a5bab701013b2bcf9adc8cd0dd100a7 100644 --- a/mindinsight/backend/profiler/profile_api.py +++ b/mindinsight/backend/profiler/profile_api.py @@ -21,7 +21,6 @@ import json import os from flask import Blueprint -from flask import Response from flask import jsonify from flask import request from marshmallow import ValidationError @@ -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.proposer.compose_proposer import ComposeProposal 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) @@ -320,7 +320,7 @@ def get_profile_summary_proposal(): proposal_obj = ComposeProposal(profiler_dir_abs, device_id, proposal_type_list) proposal_info = proposal_obj.get_proposal(options) # 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"])