Fix flaky faiss integration tests (#1635) (#1637)

(cherry picked from commit 730b7e4) Signed-off-by: Naveen Tatikonda <[email protected]> Co-authored-by: Naveen Tatikonda <[email protected]>
opensearch-project · Apr 22, 2024 · 579bbb1 · 579bbb1
1 parent 529c5de
commit 579bbb1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/test/java/org/opensearch/knn/index/FaissIT.java b/src/test/java/org/opensearch/knn/index/FaissIT.java
@@ -232,7 +232,7 @@ public void testEndToEnd_whenMethodIsHNSWFlatAndHasDeletedDocs_thenSucceed() {
 
         final Set<Integer> docIdsToBeDeleted = new HashSet<>();
         while (docIdsToBeDeleted.size() < 10) {
-            docIdsToBeDeleted.add(randomInt(testData.indexData.docs.length));
+            docIdsToBeDeleted.add(randomInt(testData.indexData.docs.length - 1));
         }
 
         for (Integer id : docIdsToBeDeleted) {
@@ -741,6 +741,7 @@ public void testIVFSQFP16_whenIndexedAndQueried_thenSucceed() {
             .field(KNN_ENGINE, FAISS_NAME)
             .field(METHOD_PARAMETER_SPACE_TYPE, "l2")
             .startObject(PARAMETERS)
+            .field(METHOD_PARAMETER_NPROBES, 4)
             .startObject(METHOD_ENCODER_PARAMETER)
             .field(NAME, ENCODER_SQ)
             .startObject(PARAMETERS)