apply spotless add comment

Signed-off-by: Brian Flores <[email protected]>
brianf-aws · Feb 10, 2025 · 6b193be · 6b193be
1 parent 95ecd22
commit 6b193be
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/plugin/src/main/java/org/opensearch/ml/action/undeploy/TransportUndeployModelsAction.java b/plugin/src/main/java/org/opensearch/ml/action/undeploy/TransportUndeployModelsAction.java
@@ -202,14 +202,16 @@ private void undeployModels(
              */
             boolean modelNotFoundInNodesCache = response.getNodes().stream().allMatch(nodeResponse -> {
                 Map<String, String> status = nodeResponse.getModelUndeployStatus();
-                if (status == null) return false;
-                boolean modelCacheMissForModelIds =  Arrays.stream(modelIds).allMatch(modelId -> {
+                if (status == null)
+                    return false;
+                // Stream is used to catch all models edge case but only one is ever undeployed
+                boolean modelCacheMissForModelIds = Arrays.stream(modelIds).allMatch(modelId -> {
                     String modelStatus = status.get(modelId);
                     return modelStatus != null && modelStatus.equalsIgnoreCase(NOT_FOUND);
                 });
 
                 return modelCacheMissForModelIds;
-            } );
+            });
             if (response.getNodes().isEmpty() || modelNotFoundInNodesCache) {
                 bulkSetModelIndexToUndeploy(modelIds, listener, response);
                 return;