From 8721079082a7531650bb7fdf5a0c359e670cae4d Mon Sep 17 00:00:00 2001
From: Varun Jain <varunudr@amazon.com>
Date: Fri, 29 Dec 2023 15:16:33 -0800
Subject: [PATCH] Fixing Test cases

Signed-off-by: Varun Jain <varunudr@amazon.com>
---
 .../java/org/opensearch/neuralsearch/bwc/SemanticSearchIT.java   | 1 +
 1 file changed, 1 insertion(+)

diff --git a/qa/restart-upgrade/src/test/java/org/opensearch/neuralsearch/bwc/SemanticSearchIT.java b/qa/restart-upgrade/src/test/java/org/opensearch/neuralsearch/bwc/SemanticSearchIT.java
index d2c0b0c12..f2b6088ee 100644
--- a/qa/restart-upgrade/src/test/java/org/opensearch/neuralsearch/bwc/SemanticSearchIT.java
+++ b/qa/restart-upgrade/src/test/java/org/opensearch/neuralsearch/bwc/SemanticSearchIT.java
@@ -42,6 +42,7 @@ public void testSemanticSearch() throws Exception {
             Map<String, Object> pipeline = getIngestionPipeline(PIPELINE_NAME);
             assertNotNull(pipeline);
             String modelId = getModelId(pipeline, TEXT_EMBEDDING_PROCESSOR);
+            loadModel(modelId);
             addDocument(getIndexNameForTest(), "1", TEST_FIELD, TEXT_1);
             validateTestIndex(modelId);
             deletePipeline(PIPELINE_NAME);