fix delete model API (#748)

Signed-off-by: Bhavana Ramaram <rbhavna@amazon.com>
opensearch-project · Mar 4, 2023 · ea67a37 · ea67a37
1 parent 6f621b6
commit ea67a37
Showing 1 changed file with 2 additions and 2 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
@@ -78,7 +78,7 @@ protected void doExecute(Task task, ActionRequest request, ActionListener<Delete
         GetRequest getRequest = new GetRequest(ML_MODEL_INDEX).id(modelId).fetchSourceContext(fetchSourceContext);
 
         try (ThreadContext.StoredContext context = client.threadPool().getThreadContext().stashContext()) {
-            client.get(getRequest, ActionListener.runBefore(ActionListener.wrap(r -> {
+            client.get(getRequest, ActionListener.wrap(r -> {
                 if (r != null && r.isExists()) {
                     try (XContentParser parser = createXContentParserFromRegistry(xContentRegistry, r.getSourceAsBytesRef())) {
                         ensureExpectedToken(XContentParser.Token.START_OBJECT, parser.nextToken(), parser);
@@ -114,7 +114,7 @@ public void onFailure(Exception e) {
                         actionListener.onFailure(e);
                     }
                 }
-            }, e -> { actionListener.onFailure(new MLResourceNotFoundException("Fail to find model")); }), () -> context.restore()));
+            }, e -> { actionListener.onFailure(new MLResourceNotFoundException("Fail to find model")); }));
         } catch (Exception e) {
             log.error("Failed to delete ML model " + modelId, e);
             actionListener.onFailure(e);