Make sure we use our Response class (#1314)

SeldonIO · Jul 25, 2023 · 4e04d1f · 4e04d1f
1 parent fa981d8
commit 4e04d1f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mlserver/rest/app.py b/mlserver/rest/app.py
@@ -28,12 +28,14 @@ def __init__(
         *args,
         response_model_exclude_unset=True,
         response_model_exclude_none=True,
+        response_class=Response,
         **kwargs
     ):
         super().__init__(
             *args,
             response_model_exclude_unset=response_model_exclude_unset,
             response_model_exclude_none=response_model_exclude_none,
+            response_class=Response,
             **kwargs
         )