From 20c8e7d763cc094ba6e88e90a8d3b68358de7aa1 Mon Sep 17 00:00:00 2001
From: VIKASH TIWARI <viktari@amazon.com>
Date: Fri, 2 Aug 2024 22:58:51 -0700
Subject: [PATCH] Bug Fix with Sampler

---
 .../opensearch/knn/quantization/sampler/ReservoirSampler.java  | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/main/java/org/opensearch/knn/quantization/sampler/ReservoirSampler.java b/src/main/java/org/opensearch/knn/quantization/sampler/ReservoirSampler.java
index 97559a3b4..0f09854ff 100644
--- a/src/main/java/org/opensearch/knn/quantization/sampler/ReservoirSampler.java
+++ b/src/main/java/org/opensearch/knn/quantization/sampler/ReservoirSampler.java
@@ -78,8 +78,7 @@ private int[] reservoirSampleIndices(final int numVectors, final int sampleSize)
                 indices[j] = i;
             }
         }
-        // If sorted indices are required, uncomment the following line
-        // Arrays.sort(indices);
+        Arrays.sort(indices);
         return indices;
     }
 }
\ No newline at end of file