Fixed tests after rebase on recent main

Signed-off-by: Martin Gaievski <[email protected]>
opensearch-project · Aug 30, 2023 · 67e19ca · 67e19ca
1 parent 614be2c
commit 67e19ca
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/test/java/org/opensearch/neuralsearch/processor/ScoreCombinationIT.java b/src/test/java/org/opensearch/neuralsearch/processor/ScoreCombinationIT.java
@@ -115,7 +115,7 @@ public void testArithmeticWeightedMean_whenWeightsPassed_thenSuccessful() {
             Map.of("search_pipeline", SEARCH_PIPELINE)
         );
 
-        assertWeightedScores(searchResponseWithWeights1AsMap, 0.375, 0.3125, 0.001);
+        assertWeightedScores(searchResponseWithWeights1AsMap, 0.4, 0.3, 0.001);
 
         // delete existing pipeline and create a new one with another set of weights
         deleteSearchPipeline(SEARCH_PIPELINE);
@@ -134,7 +134,7 @@ public void testArithmeticWeightedMean_whenWeightsPassed_thenSuccessful() {
             Map.of("search_pipeline", SEARCH_PIPELINE)
         );
 
-        assertWeightedScores(searchResponseWithWeights2AsMap, 0.606, 0.242, 0.001);
+        assertWeightedScores(searchResponseWithWeights2AsMap, 0.6666, 0.2332, 0.001);
 
         // check case when number of weights is less than number of sub-queries
         // delete existing pipeline and create a new one with another set of weights