@@ -61,7 +61,6 @@ private enum VectorSourceOptions {
6161 .collect (Collectors .toSet ());
6262
6363 public static final float DELTA = 1e-7F ;
64- public static final float BFLOAT16_DELTA = 1e-2F ;
6564
6665 private final ElementType elementType ;
6766 private final DenseVectorFieldMapper .VectorSimilarity similarity ;
@@ -71,7 +70,7 @@ private enum VectorSourceOptions {
7170 @ ParametersFactory
7271 public static Iterable <Object []> parameters () throws Exception {
7372 List <Object []> params = new ArrayList <>();
74- for (ElementType elementType : List .of (ElementType .BYTE , ElementType .FLOAT , ElementType .BIT , ElementType . BFLOAT16 )) {
73+ for (ElementType elementType : List .of (ElementType .BYTE , ElementType .FLOAT , ElementType .BIT )) {
7574 // Test all similarities
7675 for (DenseVectorFieldMapper .VectorSimilarity similarity : DenseVectorFieldMapper .VectorSimilarity .values ()) {
7776 if (elementType == ElementType .BIT && similarity != DenseVectorFieldMapper .VectorSimilarity .L2_NORM ) {
@@ -138,10 +137,8 @@ public void testRetrieveTopNDenseVectorFieldData() {
138137 } else {
139138 assertNotNull (actualVector );
140139 assertEquals (expectedVector .size (), actualVector .size ());
141-
142- float delta = elementType == ElementType .BFLOAT16 ? BFLOAT16_DELTA : DELTA ;
143140 for (int i = 0 ; i < expectedVector .size (); i ++) {
144- assertEquals (expectedVector .get (i ).floatValue (), actualVector .get (i ).floatValue (), delta );
141+ assertEquals (expectedVector .get (i ).floatValue (), actualVector .get (i ).floatValue (), DELTA );
145142 }
146143 }
147144 });
@@ -170,14 +167,12 @@ public void testRetrieveDenseVectorFieldData() {
170167 } else {
171168 assertNotNull (actualVector );
172169 assertEquals (expectedVector .size (), actualVector .size ());
173-
174- float delta = elementType == ElementType .BFLOAT16 ? BFLOAT16_DELTA : DELTA ;
175170 for (int i = 0 ; i < actualVector .size (); i ++) {
176171 assertEquals (
177172 "Actual: " + actualVector + "; expected: " + expectedVector ,
178173 expectedVector .get (i ).floatValue (),
179174 actualVector .get (i ).floatValue (),
180- delta
175+ DELTA
181176 );
182177 }
183178 }
@@ -258,13 +253,12 @@ public void setup() throws IOException {
258253 } else {
259254 for (int j = 0 ; j < numDims ; j ++) {
260255 switch (elementType ) {
261- case FLOAT , BFLOAT16 -> vector .add (randomFloatBetween (0F , 1F , true ));
256+ case FLOAT -> vector .add (randomFloatBetween (0F , 1F , true ));
262257 case BYTE , BIT -> vector .add ((byte ) randomIntBetween (-128 , 127 ));
263258 default -> throw new IllegalArgumentException ("Unexpected element type: " + elementType );
264259 }
265260 }
266- if ((elementType == ElementType .FLOAT || elementType == ElementType .BFLOAT16 )
267- && (similarity == DenseVectorFieldMapper .VectorSimilarity .DOT_PRODUCT || rarely ())) {
261+ if ((elementType == ElementType .FLOAT ) && (similarity == DenseVectorFieldMapper .VectorSimilarity .DOT_PRODUCT || rarely ())) {
268262 // Normalize the vector
269263 float magnitude = DenseVector .getMagnitude (vector );
270264 vector .replaceAll (number -> number .floatValue () / magnitude );
0 commit comments