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]>
  • Loading branch information
ylwu-amzn authored Oct 11, 2023
1 parent 5544681 commit cea1cd6
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 cea1cd6

Please sign in to comment.