Bug fix of load model id

Signed-off-by: Varun Jain <[email protected]>
opensearch-project · Jan 2, 2024 · 136bb81 · 136bb81
1 parent 7f3e9e4
commit 136bb81
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/qa/rolling-upgrade/src/test/java/org/opensearch/neuralsearch/bwc/SemanticSearchIT.java b/qa/rolling-upgrade/src/test/java/org/opensearch/neuralsearch/bwc/SemanticSearchIT.java
@@ -54,6 +54,7 @@ public void testSemanticSearch_E2EFlow() throws Exception {
             case UPGRADED:
                 modelId = getModelId(PIPELINE_NAME);
                 int totalDocsCountUpgraded = 3 * NUM_DOCS_PER_ROUND;
+                loadModel(modelId);
                 addDocument(getIndexNameForTest(), "2", TEST_FIELD, TEXT_UPGRADED);
                 validateTestIndexOnUpgrade(totalDocsCountUpgraded, modelId, TEXT_UPGRADED);
                 deletePipeline(PIPELINE_NAME);