From eaf357ed16e39c665b642683bcbb1c178e92e8b9 Mon Sep 17 00:00:00 2001
From: Håvard Pettersen <havardpe@oath.com>
Date: Tue, 26 Nov 2019 13:15:04 +0000
Subject: simplify

---
 .../queryeval/nearest_neighbor/nearest_neighbor_test.cpp     | 12 ++----------
 1 file changed, 2 insertions(+), 10 deletions(-)

(limited to 'searchlib/src/tests/queryeval/nearest_neighbor/nearest_neighbor_test.cpp')

diff --git a/searchlib/src/tests/queryeval/nearest_neighbor/nearest_neighbor_test.cpp b/searchlib/src/tests/queryeval/nearest_neighbor/nearest_neighbor_test.cpp
index 8146bdfeaa5..6a96b7720b1 100644
--- a/searchlib/src/tests/queryeval/nearest_neighbor/nearest_neighbor_test.cpp
+++ b/searchlib/src/tests/queryeval/nearest_neighbor/nearest_neighbor_test.cpp
@@ -143,19 +143,11 @@ std::vector<feature_t> get_rawscores(Fixture &env, const DenseTensorView &qtv) {
     search->initRange(docid, limit);
     std::vector<feature_t> rv;
     while (docid < limit) {
-        if (strict) {
-            search->seek(docid);
-            if (search->isAtEnd()) break;
-            docid = search->getDocId();
+        if (search->seek(docid)) {
             search->unpack(docid);
             rv.push_back(tfmd.getRawScore());
-        } else {
-            if (search->seek(docid)) {
-                search->unpack(docid);
-                rv.push_back(tfmd.getRawScore());
-            }
         }
-        ++docid;
+        docid = std::max(search->getDocId(), docid + 1);
     }
     return rv;
 }
-- 
cgit v1.2.3