Save number of points loaded in the label file for checking later

microsoft · Oct 1, 2024 · 532ab2f · 532ab2f
1 parent a5f3d2e
commit 532ab2f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/in_mem_filter_store.cpp b/src/in_mem_filter_store.cpp
@@ -193,6 +193,8 @@ namespace diskann {
     get_label_file_metadata(label_file_content, num_pts_in_label_file,
                             num_total_labels);
 
+    _num_points = num_pts_in_label_file;
+
     _pts_to_label_offsets = new uint32_t[num_pts_in_label_file];
     _pts_to_label_counts = new uint32_t[num_pts_in_label_file];
     _pts_to_labels = new LabelT[num_total_labels];