fix test

opensearch-project · Jan 8, 2025 · 161307d · 161307d
1 parent b151c5c
commit 161307d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/qa/rolling-upgrade/src/test/java/org/opensearch/neuralsearch/bwc/HybridSearchIT.java b/qa/rolling-upgrade/src/test/java/org/opensearch/neuralsearch/bwc/HybridSearchIT.java
@@ -80,7 +80,7 @@ public void testNormalizationProcessor_whenIndexWithMultipleShards_E2EFlow() thr
             case UPGRADED:
                 try {
                     modelId = getModelId(getIngestionPipeline(PIPELINE_NAME), TEXT_EMBEDDING_PROCESSOR);
-                    int totalDocsCountUpgraded = 1 * NUM_DOCS_PER_ROUND;
+                    int totalDocsCountUpgraded = 2 * NUM_DOCS_PER_ROUND;
                     loadModel(modelId);
                     addDocument(getIndexNameForTest(), "1", TEST_FIELD, TEXT_UPGRADED, null, null);
                     HybridQueryBuilder hybridQueryBuilder = getQueryBuilder(modelId, null, null, null);