From bfb6b7f994bd37aacf3cb1b6e002a80646096f38 Mon Sep 17 00:00:00 2001
From: jzonthemtn <jeff.zemerick@mtnfog.com>
Date: Tue, 21 Jan 2025 08:30:44 -0500
Subject: [PATCH] Fixing query size with >=.

---
 .../org/opensearch/eval/samplers/AllQueriesQuerySampler.java    | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/main/java/org/opensearch/eval/samplers/AllQueriesQuerySampler.java b/src/main/java/org/opensearch/eval/samplers/AllQueriesQuerySampler.java
index 8d6e61a..54a78b5 100644
--- 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;
             }