Skip to content

Commit

Permalink
fix for sync up job not working in 2.17 when upgraded from previous v…
Browse files Browse the repository at this point in the history
…ersions

Signed-off-by: Bhavana Goud Ramaram <[email protected]>
  • Loading branch information
rbhavna committed Nov 27, 2024
1 parent 78a304a commit 1af10ae
Showing 1 changed file with 4 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -261,6 +261,7 @@ private void queryRunningModels(ActionListener<SearchResponse> listener) {
private void triggerModelRedeploy(ModelAutoRedeployArrangement modelAutoRedeployArrangement) {
if (modelAutoRedeployArrangement == null) {
log.info("No more models in arrangement, skipping the redeployment");
redeployAModel();
return;
}
String modelId = modelAutoRedeployArrangement.getSearchResponse().getId();
Expand All @@ -275,10 +276,12 @@ private void triggerModelRedeploy(ModelAutoRedeployArrangement modelAutoRedeploy
"Model function_name or algorithm is null, model is not in correct status, please check the model, model id is: {}",
modelId
);
redeployAModel();
return;
}
if (FunctionName.REMOTE == FunctionName.from(functionName)) {
log.info("Skipping redeploying remote model {} as remote model deployment can be done at prediction time.", modelId);
redeployAModel();
return;
}
List<String> planningWorkerNodes = (List<String>) sourceAsMap.get(MLModel.PLANNING_WORKER_NODES_FIELD);
Expand All @@ -302,6 +305,7 @@ private void triggerModelRedeploy(ModelAutoRedeployArrangement modelAutoRedeploy
.info(
"Allow custom deployment plan is true and deploy to all nodes is false and added nodes are not in planning worker nodes list, not to auto redeploy the model to the new nodes!"
);
redeployAModel();
return;
}

Expand Down

0 comments on commit 1af10ae

Please sign in to comment.