Fix BQVectors#ramBytesUsed and BQVectors#getCompressedSize on BQVecto…

…rs instance that is initialized but has no vectors (such as a fresh MutableBQVectors instance)
jbellis · Feb 13, 2025 · 219465f · 219465f
1 parent 188c216
commit 219465f
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/jvector-base/src/main/java/io/github/jbellis/jvector/quantization/BQVectors.java b/jvector-base/src/main/java/io/github/jbellis/jvector/quantization/BQVectors.java
@@ -119,7 +119,7 @@ public int getOriginalSize() {
 
     @Override
     public int getCompressedSize() {
-        return compressedVectors[0].length * Long.BYTES;
+        return bq.compressedVectorSize();
     }
 
     @Override
@@ -129,7 +129,11 @@ public BinaryQuantization getCompressor() {
 
     @Override
     public long ramBytesUsed() {
-        return count() * RamUsageEstimator.sizeOf(compressedVectors[0]);
+        long[] compressedVector = compressedVectors[0];
+        if (compressedVector == null) {
+            return 0;
+        }
+        return count() * RamUsageEstimator.sizeOf(compressedVector);
     }
 
     @Override