fix no worker node error on multi-node cluster (opensearch-project#1487…

…) (opensearch-project#1513) Signed-off-by: Yaliang Wu <[email protected]> (cherry picked from commit cea1cd6) Co-authored-by: Yaliang Wu <[email protected]>
austintlee · Feb 29, 2024 · 7b76e78 · 7b76e78
1 parent 0bb17ac
commit 7b76e78
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/plugin/src/main/java/org/opensearch/ml/action/prediction/TransportPredictionTaskAction.java b/plugin/src/main/java/org/opensearch/ml/action/prediction/TransportPredictionTaskAction.java
@@ -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);
             }
         }