add log for corner case

Signed-off-by: Yaliang Wu <[email protected]>
opensearch-project · Jan 6, 2023 · f60641a · f60641a
1 parent edf0732
commit f60641a
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/plugin/src/main/java/org/opensearch/ml/action/syncup/TransportSyncUpOnNodeAction.java b/plugin/src/main/java/org/opensearch/ml/action/syncup/TransportSyncUpOnNodeAction.java
@@ -204,6 +204,16 @@ void cleanUpLocalCache() {
                         modelState = MLModelState.PARTIALLY_LOADED;
                     } else {
                         modelState = MLModelState.LOADED;
+                        if (mlTask.getWorkerNodes().size() < workerNodes.length) {
+                            log
+                                .warn(
+                                    "Model loaded on more nodes than target worker nodes. taskId:{}, modelId: {}, workerNodes: {}, targetWorkerNodes: {}",
+                                    taskId,
+                                    modelId,
+                                    Arrays.toString(workerNodes),
+                                    Arrays.toString(mlTask.getWorkerNodes().toArray(new String[0]))
+                                );
+                        }
                     }
                     log.info("Reset model state as {} for model {}", modelState, modelId);
                     mlModelManager.updateModel(modelId, ImmutableMap.of(MLModel.MODEL_STATE_FIELD, modelState));