add status to response

SeldonIO · Oct 18, 2019 · 2ff7329 · 2ff7329
1 parent 4fc71c2
commit 2ff7329
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/python/seldon_core/wrapper.py b/python/seldon_core/wrapper.py
@@ -40,8 +40,12 @@ def Predict():
         requestJson = get_request()
         logger.debug("REST Request: %s", request)
         response = seldon_core.seldon_methods.predict(user_model, requestJson)
+        json_response = jsonify(response)
+        if response.status is not None and response.status.code is not None:
+            json_response.status_code = response.status.code
+
         logger.debug("REST Response: %s", response)
-        return jsonify(response)
+        return json_response
 
     @app.route("/send-feedback", methods=["GET", "POST"])
     @app.route("/api/v0.1/feedback", methods=["POST"])