From 08ac4c0f6a819f9cc92a5942f310d8652ad6b0b3 Mon Sep 17 00:00:00 2001
From: Yaliang Wu <ylwu@amazon.com>
Date: Wed, 11 Oct 2023 10:40:11 -0700
Subject: [PATCH] fix no worker node error on multi-node cluster (#1487)

Signed-off-by: Yaliang Wu <ylwu@amazon.com>
(cherry picked from commit cea1cd675cd95e37c29a3bd88c1cd9d58e81b20a)
---
 .../ml/action/prediction/TransportPredictionTaskAction.java    | 3 +--
 1 file changed, 1 insertion(+), 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
index 1d58a5d0db..63be5e2423 100644
--- 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);
             }
         }