Fix delete training model cleanup in bwc

Signed-off-by: John Mazanec <[email protected]>

qa/restart-upgrade/src/test/java/org/opensearch/knn/bwc/ModelIT.java

@@ -21,14 +21,12 @@
 import org.opensearch.knn.indices.ModelMetadata;
 import org.opensearch.knn.indices.ModelState;
 import org.opensearch.knn.plugin.KNNPlugin;
-import org.opensearch.knn.plugin.transport.DeleteModelResponse;
 import org.opensearch.rest.RestStatus;
 import org.opensearch.search.SearchHit;
 
 import java.io.IOException;
 import java.nio.charset.StandardCharsets;
 import java.util.Arrays;
-import java.util.Locale;
 import java.util.Map;
 
 import static org.opensearch.knn.TestUtils.KNN_BWC_PREFIX;
@@ -156,6 +154,10 @@ public void testDeleteTrainingModel() throws Exception {
 
             ResponseException ex = expectThrows(ResponseException.class, () -> client().performRequest(request));
             assertEquals(RestStatus.CONFLICT.getStatus(), ex.getResponse().getStatusLine().getStatusCode());
+
+            Request deleteModeFromIndexRequest = new Request("DELETE", String.join("/", MODEL_INDEX_NAME, TEST_MODEL_ID_TRAINING));
+            Response r = client().performRequest(deleteModeFromIndexRequest);
+            assertEquals(RestStatus.OK.getStatus(), r.getStatusLine().getStatusCode());
         }
     }
 
@@ -165,7 +167,6 @@ public static void wipeAllModels() throws IOException {
         if (!isRunningAgainstOldCluster()) {
             deleteKNNModel(TEST_MODEL_ID);
             deleteKNNModel(TEST_MODEL_ID_DEFAULT);
-            deleteKNNModel(TEST_MODEL_ID_TRAINING);
 
             Request request = new Request("DELETE", "/" + MODEL_INDEX_NAME);