Fixing query size with >=.

o19s · Jan 21, 2025 · bfb6b7f · bfb6b7f
1 parent b58b3fe
commit bfb6b7f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main/java/org/opensearch/eval/samplers/AllQueriesQuerySampler.java b/src/main/java/org/opensearch/eval/samplers/AllQueriesQuerySampler.java
@@ -51,7 +51,7 @@ public String sample() throws Exception {
             queries.merge(ubiQuery.getUserQuery(), 1L, Long::sum);
 
             // Will be useful for paging once implemented.
-            if(queries.size() > parameters.getQuerySetSize()) {
+            if(queries.size() >= parameters.getQuerySetSize()) {
                 break;
             }