Fix flaky faiss integration tests (opensearch-project#1635)

peterzhuamazon · Apr 20, 2024 · 730b7e4 · 730b7e4
1 parent a2bb6cc
commit 730b7e4
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)