fix reindex issue

Signed-off-by: Bhavana Ramaram <rbhavna@amazon.com>
rbhavna · Oct 6, 2023 · c32b576 · c32b576
1 parent 94f590f
commit c32b576
Showing 1 changed file with 11 additions and 13 deletions.
diff --git a/plugin/src/main/java/org/opensearch/ml/action/models/DeleteModelTransportAction.java b/plugin/src/main/java/org/opensearch/ml/action/models/DeleteModelTransportAction.java
@@ -5,13 +5,10 @@
 
 package org.opensearch.ml.action.models;
 
-import static org.opensearch.core.xcontent.XContentParserUtils.ensureExpectedToken;
-import static org.opensearch.ml.common.CommonValue.ML_MODEL_INDEX;
-import static org.opensearch.ml.common.MLModel.ALGORITHM_FIELD;
-import static org.opensearch.ml.common.MLModel.MODEL_ID_FIELD;
-import static org.opensearch.ml.utils.MLNodeUtils.createXContentParserFromRegistry;
-import static org.opensearch.ml.utils.RestActionUtils.getFetchSourceContext;
-
+import com.google.common.annotations.VisibleForTesting;
+import lombok.AccessLevel;
+import lombok.experimental.FieldDefaults;
+import lombok.extern.log4j.Log4j2;
 import org.opensearch.OpenSearchStatusException;
 import org.opensearch.ResourceNotFoundException;
 import org.opensearch.action.ActionRequest;
@@ -46,11 +43,12 @@
 import org.opensearch.tasks.Task;
 import org.opensearch.transport.TransportService;
 
-import com.google.common.annotations.VisibleForTesting;
-
-import lombok.AccessLevel;
-import lombok.experimental.FieldDefaults;
-import lombok.extern.log4j.Log4j2;
+import static org.opensearch.core.xcontent.XContentParserUtils.ensureExpectedToken;
+import static org.opensearch.ml.common.CommonValue.ML_MODEL_INDEX;
+import static org.opensearch.ml.common.MLModel.ALGORITHM_FIELD;
+import static org.opensearch.ml.common.MLModel.MODEL_ID_FIELD;
+import static org.opensearch.ml.utils.MLNodeUtils.createXContentParserFromRegistry;
+import static org.opensearch.ml.utils.RestActionUtils.getFetchSourceContext;
 
 @Log4j2
 @FieldDefaults(level = AccessLevel.PRIVATE)
@@ -118,7 +116,7 @@ protected void doExecute(Task task, ActionRequest request, ActionListener<Delete
                                     || mlModelState.equals(MLModelState.DEPLOYED)
                                     || mlModelState.equals(MLModelState.DEPLOYING)
                                     || mlModelState.equals(MLModelState.PARTIALLY_DEPLOYED)) {
-                                    actionListener
+                                    wrappedListener
                                         .onFailure(
                                             new Exception(
                                                 "Model cannot be deleted in deploying or deployed state. Try undeploy model first then delete"