Tests passing locally

alexklibisz · Aug 26, 2024 · 56348f2 · 56348f2
1 parent 1e008ba
commit 56348f2
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 18 deletions.
diff --git a/elastiknn-lucene/src/main/java/com/klibisz/elastiknn/search/ArrayHitCounter.java b/elastiknn-lucene/src/main/java/com/klibisz/elastiknn/search/ArrayHitCounter.java
@@ -119,16 +119,16 @@ public int nextDoc() {
                     while (true) {
                         if (numEmitted == candidates || docID + 1 > maxKey) {
                             docID = DocIdSetIterator.NO_MORE_DOCS;
-                            return docID();
+                            return docID;
                         } else {
                             docID++;
                             if (counts[docID] > kgr.kthGreatest) {
                                 numEmitted++;
-                                return docID();
+                                return docID;
                             } else if (counts[docID] == kgr.kthGreatest && numEq < candidates - kgr.numGreaterThan) {
                                 numEq++;
                                 numEmitted++;
-                                return docID();
+                                return docID;
                             }
                         }
                     }

diff --git a/elastiknn-plugin-integration-tests/src/test/scala/com/klibisz/elastiknn/RecallSuite.scala b/elastiknn-plugin-integration-tests/src/test/scala/com/klibisz/elastiknn/RecallSuite.scala
@@ -40,21 +40,21 @@ class RecallSuite extends AsyncFunSuite with Matchers with ElasticAsyncClient wi
 
   private val tests = Seq(
     // Exact
-//    Test(
-//      Mapping.SparseBool(dims),
-//      Seq(
-//        NearestNeighborsQuery.Exact(vecField, Similarity.Jaccard) -> 1d,
-//        NearestNeighborsQuery.Exact(vecField, Similarity.Hamming) -> 1d
-//      )
-//    ),
-//    Test(
-//      Mapping.DenseFloat(dims),
-//      Seq(
-//        NearestNeighborsQuery.Exact(vecField, Similarity.L1) -> 1d,
-//        NearestNeighborsQuery.Exact(vecField, Similarity.L2) -> 1d,
-//        NearestNeighborsQuery.Exact(vecField, Similarity.Cosine) -> 1d
-//      )
-//    ),
+    Test(
+      Mapping.SparseBool(dims),
+      Seq(
+        NearestNeighborsQuery.Exact(vecField, Similarity.Jaccard) -> 1d,
+        NearestNeighborsQuery.Exact(vecField, Similarity.Hamming) -> 1d
+      )
+    ),
+    Test(
+      Mapping.DenseFloat(dims),
+      Seq(
+        NearestNeighborsQuery.Exact(vecField, Similarity.L1) -> 1d,
+        NearestNeighborsQuery.Exact(vecField, Similarity.L2) -> 1d,
+        NearestNeighborsQuery.Exact(vecField, Similarity.Cosine) -> 1d
+      )
+    ),
     // Jaccard LSH
     Test(
       Mapping.JaccardLsh(dims, 200, 1),