awslabs / multi-model-server

Multi Model Server is a tool for serving neural net models for inference
Apache License 2.0
998 stars 230 forks source link

MMS - Support for custom error codes in custom handlers #965

Closed dhanainme closed 3 years ago

dhanainme commented 3 years ago

Fixes - Allow custom HTTP status in mms.service.Service #961

Description of changes:

Defines a custom exception mms.service.PredictionException enabling return of custom error codes

Testing done:

Added unit tests / archive to test & verify the behavior in the PR.

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.