Clear planningWorkerNodes when model auto-deploys again after undeploy (

opensearch-project#2396) Signed-off-by: Xun Zhang <xunzh@amazon.com>
zane-neo · May 1, 2024 · 8cb6a9b · 8cb6a9b
1 parent ceb6b16
commit 8cb6a9b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/plugin/src/main/java/org/opensearch/ml/task/MLPredictTaskRunner.java b/plugin/src/main/java/org/opensearch/ml/task/MLPredictTaskRunner.java
@@ -167,6 +167,10 @@ public void dispatchTask(
                                 return;
                             }
                             String[] planningWorkerNodes = model.getPlanningWorkerNodes();
+                            boolean deployToAllNodes = model.isDeployToAllNodes();
+                            if (deployToAllNodes) {
+                                planningWorkerNodes = null;
+                            }
                             MLModel modelBeingAutoDeployed = mlModelManager.addModelToAutoDeployCache(modelId, model);
                             if (modelBeingAutoDeployed == model) {
                                 log.info(getErrorMessage("Automatically deploy model", modelId, isHidden));