fix integration test

Signed-off-by: Hailong Cui <ihailong@amazon.com>
opensearch-project · Apr 26, 2024 · b7f7a03 · b7f7a03
1 parent e25707d
commit b7f7a03
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/plugin/src/test/java/org/opensearch/ml/tools/ToolIntegrationWithLLMTest.java b/plugin/src/test/java/org/opensearch/ml/tools/ToolIntegrationWithLLMTest.java
@@ -62,6 +62,7 @@ public void stopMockLLM() {
 
     @After
     public void deleteModel() throws IOException {
+        undeployModel(modelId);
         deleteModel(client(), modelId, null);
     }