fix no worker node error on multi-node cluster (#1487) (#1513)

Signed-off-by: Yaliang Wu <[email protected]> (cherry picked from commit cea1cd6) Co-authored-by: Yaliang Wu <[email protected]>
opensearch-project · Nov 20, 2023 · 78f61ca · 78f61ca
1 parent a58d8ea
commit 78f61ca
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);
             }
         }