Skip to content

Commit

Permalink
fix no worker node error on multi-node cluster (#1487)
Browse files Browse the repository at this point in the history
Signed-off-by: Yaliang Wu <[email protected]>
(cherry picked from commit cea1cd6)
  • Loading branch information
ylwu-amzn authored and github-actions[bot] committed Oct 11, 2023
1 parent 60ef0fd commit 435597b
Showing 1 changed file with 1 addition and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -121,9 +121,8 @@ public void onFailure(Exception e) {

if (cachedMlModel != null) {
modelActionListener.onResponse(cachedMlModel);
} else if (modelAccessControlHelper.skipModelAccessControl(user)) {
executePredict(mlPredictionTaskRequest, wrappedListener, modelId);
} else {
// For multi-node cluster, the function name is null in cache, so should always get model first.
mlModelManager.getModel(modelId, modelActionListener);
}
}
Expand Down

0 comments on commit 435597b

Please sign in to comment.