From c31d213fdde32a28585f660204b7234df3941a70 Mon Sep 17 00:00:00 2001 From: shamika Date: Tue, 7 Mar 2023 23:23:08 +0000 Subject: [PATCH 1/2] Handled error to return the reason as part of the reponse. --- mms/service.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/mms/service.py b/mms/service.py index 1ebe47cc0..96ef430f9 100644 --- a/mms/service.py +++ b/mms/service.py @@ -109,12 +109,12 @@ def predict(self, batch): except PredictionException as e: logger.error("Prediction error", exc_info=True) return create_predict_response(None, req_id_map, e.message, e.error_code) - except MemoryError: + except MemoryError as e: logger.error("System out of memory", exc_info=True) return create_predict_response(None, req_id_map, "Out of resources", 507) - except Exception: # pylint: disable=broad-except + except Exception as e: # pylint: disable=broad-except logger.warning("Invoking custom service failed.", exc_info=True) - return create_predict_response(None, req_id_map, "Prediction failed", 503) + return create_predict_response(None, req_id_map, "Prediction failed : [{}]".format(e), 503) if not isinstance(ret, list): logger.warning("model: %s, Invalid return type: %s.", self.context.model_name, type(ret)) From 860883ce87eba393373eeda130bc01829336a33e Mon Sep 17 00:00:00 2001 From: shamika Date: Tue, 7 Mar 2023 23:26:12 +0000 Subject: [PATCH 2/2] Handled error to return the reason as part of the reponse. --- mms/service.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mms/service.py b/mms/service.py index 96ef430f9..8f7c0bc95 100644 --- a/mms/service.py +++ b/mms/service.py @@ -109,7 +109,7 @@ def predict(self, batch): except PredictionException as e: logger.error("Prediction error", exc_info=True) return create_predict_response(None, req_id_map, e.message, e.error_code) - except MemoryError as e: + except MemoryError: logger.error("System out of memory", exc_info=True) return create_predict_response(None, req_id_map, "Out of resources", 507) except Exception as e: # pylint: disable=broad-except