Wrapper fixed

SeldonIO · Aug 14, 2020 · e2a2dc0 · e2a2dc0
1 parent fb8be04
commit e2a2dc0
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/python/seldon_core/wrapper.py b/python/seldon_core/wrapper.py
@@ -179,6 +179,7 @@ def _set_flask_app_configs(app):
     env_to_config_map = {
         "FLASK_JSONIFY_PRETTYPRINT_REGULAR": "JSONIFY_PRETTYPRINT_REGULAR",
         "FLASK_JSON_SORT_KEYS": "JSON_SORT_KEYS",
+        "FLASK_PROPAGATE_EXCEPTIONS": "PROPAGATE_EXCEPTIONS",
     }
 
     for env_var, config_name in env_to_config_map.items():
@@ -198,7 +199,9 @@ def __init__(self, user_model, seldon_metrics):
         self.user_model = user_model
         self.seldon_metrics = seldon_metrics
 
-        self.metadata_data = seldon_core.seldon_methods.init_metadata(user_model)
+        self.metadata_data = seldon_core.seldon_methods.init_metadata(
+            user_model
+        )
 
     def Predict(self, request_grpc, context):
         return seldon_core.seldon_methods.predict(
@@ -240,7 +243,9 @@ def ModelMetadata(self, request_grpc, context):
 
     def GraphMetadata(self, request_grpc, context):
         """GraphMetadata method of rpc Seldon service"""
-        raise NotImplementedError("GraphMetadata not available on the Model level.")
+        raise NotImplementedError(
+            "GraphMetadata not available on the Model level."
+        )
 
 
 def get_grpc_server(