Skip to content

Commit

Permalink
Minor change
Browse files Browse the repository at this point in the history
Signed-off-by: Ryan Bogan <[email protected]>
  • Loading branch information
ryanbogan committed Sep 27, 2023
1 parent a73ddd8 commit 4faab0e
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion src/testFixtures/java/org/opensearch/knn/TestUtils.java
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@
import java.io.FileReader;
import java.io.IOException;
import org.opensearch.knn.index.SpaceType;
import org.opensearch.knn.index.codec.util.SerializationMode;
import org.opensearch.knn.plugin.script.KNNScoringUtil;
import java.util.Comparator;
import java.util.Random;
Expand Down Expand Up @@ -283,7 +284,7 @@ private KNNCodecUtil.Pair readIndexData(String path) throws IOException {
}
}

return new KNNCodecUtil.Pair(idsArray, vectorsArray);
return new KNNCodecUtil.Pair(idsArray, vectorsArray, SerializationMode.COLLECTION_OF_FLOATS);
}

private float[][] readQueries(String path) throws IOException {
Expand Down

0 comments on commit 4faab0e

Please sign in to comment.