more accurate estimate

1 files changed, 5 insertions, 1 deletions

diff --git a/searchlib/src/vespa/searchlib/queryeval/nearest_neighbor_blueprint.cpp b/searchlib/src/vespa/searchlib/queryeval/nearest_neighbor_blueprint.cpp
index 7f7c4d2976b..d4aa2aaa1d7 100644
--- a/searchlib/src/vespa/searchlib/queryeval/nearest_neighbor_blueprint.cpp
+++ b/searchlib/src/vespa/searchlib/queryeval/nearest_neighbor_blueprint.cpp
@@ -21,7 +21,11 @@ NearestNeighborBlueprint::NearestNeighborBlueprint(const queryeval::FieldSpec& f
       _distance_heap(target_num_hits),
       _found_hits()
 {
-    setEstimate(HitEstimate(_attr_tensor.getNumDocs(), false));
+    uint32_t est_hits = _attr_tensor.getNumDocs();
+    if (_attr_tensor.nearest_neighbor_index()) {
+        est_hits = std::min(target_num_hits, est_hits);
+    }
+    setEstimate(HitEstimate(est_hits, false));
 }
 
 NearestNeighborBlueprint::~NearestNeighborBlueprint() = default;