fixing tests

benwtrent · Jun 25, 2024 · e3b07dd · e3b07dd
1 parent ecc2e35
commit e3b07dd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/...er/src/test/java/org/elasticsearch/index/codec/vectors/ES815BitFlatVectorFormatTests.java b/...er/src/test/java/org/elasticsearch/index/codec/vectors/ES815BitFlatVectorFormatTests.java
@@ -11,6 +11,7 @@
 import org.apache.lucene.codecs.Codec;
 import org.apache.lucene.codecs.KnnVectorsFormat;
 import org.apache.lucene.codecs.lucene99.Lucene99Codec;
+import org.apache.lucene.index.VectorSimilarityFunction;
 import org.junit.Before;
 
 public class ES815BitFlatVectorFormatTests extends BaseKnnBitVectorsFormatTestCase {
@@ -27,7 +28,7 @@ public KnnVectorsFormat getKnnVectorsFormatForField(String field) {
 
     @Before
     public void init() {
-        similarityFunction = randomSimilarity();
+        similarityFunction = VectorSimilarityFunction.EUCLIDEAN;
     }
 
 }
diff --git a/...r/src/test/java/org/elasticsearch/index/codec/vectors/ES815HnswBitVectorsFormatTests.java b/...r/src/test/java/org/elasticsearch/index/codec/vectors/ES815HnswBitVectorsFormatTests.java
@@ -11,6 +11,7 @@
 import org.apache.lucene.codecs.Codec;
 import org.apache.lucene.codecs.KnnVectorsFormat;
 import org.apache.lucene.codecs.lucene99.Lucene99Codec;
+import org.apache.lucene.index.VectorSimilarityFunction;
 import org.junit.Before;
 
 public class ES815HnswBitVectorsFormatTests extends BaseKnnBitVectorsFormatTestCase {
@@ -27,6 +28,6 @@ public KnnVectorsFormat getKnnVectorsFormatForField(String field) {
 
     @Before
     public void init() {
-        similarityFunction = randomSimilarity();
+        similarityFunction = VectorSimilarityFunction.EUCLIDEAN;
     }
 }