Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
🚀 What
Adds a
is_ready
function to the model wrapper that users can use to define their custom health checks. This function is called on calls to this endpoint/v1/models/{model_name}
(our existing health check endpoint), if defined. Otherwise we'll fall back to the existing health check functionality which checks if either the load failed or the model is not in a ready state.🔬 Testing
TODO: Test on dev with an RC
Added a couple of integration tests and ensured they pass locally.