From af1535fee7c25766269478b34a9907beec6ededc Mon Sep 17 00:00:00 2001
From: Xuanwo <github@xuanwo.io>
Date: Tue, 27 Oct 2026 09:20:47 +0800
Subject: [PATCH] test(rust): maintain vector index in RTE geo test

---
 rust/lancedb/src/query.rs | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/rust/lancedb/src/query.rs b/rust/lancedb/src/query.rs
index 06ce17088..f30870fb1 100644
--- a/rust/lancedb/src/query.rs
+++ b/rust/lancedb/src/query.rs
@@ -3361,11 +3361,21 @@ mod tests {
     #[cfg(feature = "geo")]
     #[tokio::test]
     async fn test_st_dwithin_filter_on_rte_base_and_active_memtable() {
+        use crate::index::vector::IvfPqIndexBuilder;
+
         let tmp_dir = tempdir().unwrap();
         let table = make_geo_query_table(&tmp_dir).await;
         table.set_unenforced_primary_key(["id"]).await.unwrap();
         table
-            .enable_rte(crate::table::RTEWriteSpec::unsharded())
+            .create_index(&["vector"], Index::IvfPq(IvfPqIndexBuilder::default()))
+            .execute()
+            .await
+            .unwrap();
+        let vec_index = table.list_indices().await.unwrap()[0].name.clone();
+        table
+            .enable_rte(
+                crate::table::RTEWriteSpec::unsharded().with_maintained_indexes(vec![vec_index]),
+            )
             .await
             .unwrap();