Skip to content

Commit 937cf87

Browse files
Clear planningWorkerNodes when model auto-deploys again after undeploy (opensearch-project#2396) (opensearch-project#2397)
Signed-off-by: Xun Zhang <xunzh@amazon.com> (cherry picked from commit 8cb6a9b) Co-authored-by: Xun Zhang <xunzh@amazon.com>
1 parent 7959105 commit 937cf87

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

plugin/src/main/java/org/opensearch/ml/task/MLPredictTaskRunner.java

+4
Original file line numberDiff line numberDiff line change
@@ -167,6 +167,10 @@ public void dispatchTask(
167167
return;
168168
}
169169
String[] planningWorkerNodes = model.getPlanningWorkerNodes();
170+
boolean deployToAllNodes = model.isDeployToAllNodes();
171+
if (deployToAllNodes) {
172+
planningWorkerNodes = null;
173+
}
170174
MLModel modelBeingAutoDeployed = mlModelManager.addModelToAutoDeployCache(modelId, model);
171175
if (modelBeingAutoDeployed == model) {
172176
log.info(getErrorMessage("Automatically deploy model", modelId, isHidden));

0 commit comments

Comments
 (0)