Bug Fix

Signed-off-by: Varun Jain <[email protected]>
opensearch-project · Dec 26, 2023 · ba17401 · ba17401
1 parent 74d9aef
commit ba17401
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/qa/rolling-upgrade/src/test/java/org/opensearch/neuralsearch/bwc/SemanticSearch.java b/qa/rolling-upgrade/src/test/java/org/opensearch/neuralsearch/bwc/SemanticSearch.java
@@ -9,9 +9,8 @@
 import java.nio.file.Files;
 import java.nio.file.Path;
 import java.util.Map;
-import java.util.ArrayList;
+import org.opensearch.neuralsearch.TestUtils;
 import static org.opensearch.neuralsearch.TestUtils.NODES_BWC_CLUSTER;
-import static org.opensearch.neuralsearch.TestUtils.getModelId;
 import static org.opensearch.neuralsearch.TestUtils.TEXT_EMBEDDING_PROCESSOR;
 import org.opensearch.neuralsearch.query.NeuralQueryBuilder;
 
@@ -97,8 +96,8 @@ protected void createPipelineProcessor(String modelId, String pipelineName, Proc
     }
 
     private String getModelId(String pipelineName){
-        Map<String,Object> pipeline = getIngestionPipeline(PIPELINE_NAME);
+        Map<String,Object> pipeline = getIngestionPipeline(pipelineName);
         assertNotNull(pipeline);
-        return getModelId(pipeline,TEXT_EMBEDDING_PROCESSOR);
+        return TestUtils.getModelId(pipeline,TEXT_EMBEDDING_PROCESSOR);
     }
 }