Update IndexRefine.cpp

facebookresearch · Nov 13, 2024 · 9350e67 · 9350e67
1 parent 787cba9
commit 9350e67
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/faiss/IndexRefine.cpp b/faiss/IndexRefine.cpp
@@ -54,6 +54,14 @@ void IndexRefine::add(idx_t n, const float* x) {
     ntotal = refine_index->ntotal;
 }
 
+size_t IndexRefine::remove_ids(const IDSelector& sel) {
+    FAISS_THROW_IF_NOT(is_trained);
+    base_index->remove_ids(sel);
+    refine_index->remove_ids(sel);
+    ntotal = refine_index->ntotal;
+    return -1;
+}
+
 void IndexRefine::reset() {
     base_index->reset();
     refine_index->reset();