From b9bdb8d9374732ded7a023b479ca4aaa7a096fd8 Mon Sep 17 00:00:00 2001 From: LuQQiu Date: Sat, 29 Mar 2025 22:46:57 -0700 Subject: [PATCH] fix: fix remote restore api to always checkout latest version (#2291) Fix restore to always checkout latest version, following local restore api implementation https://github.com/lancedb/lancedb/blob/a1d1833a40eb03f83a00921279652c36e4836e9c/rust/lancedb/src/table.rs#L1910 Otherwise table.create_table -> version 1 table.add_table -> version 2 table.checkout(1), table.restore() -> the version remains at 1 (should checkout_latest inside restore method to update version to latest version and allow write operation) table.checkout_latest() -> version is 3 can do write operations --- rust/lancedb/src/remote/table.rs | 1 + 1 file changed, 1 insertion(+) diff --git a/rust/lancedb/src/remote/table.rs b/rust/lancedb/src/remote/table.rs index bae9bed2..4991f968 100644 --- a/rust/lancedb/src/remote/table.rs +++ b/rust/lancedb/src/remote/table.rs @@ -434,6 +434,7 @@ impl BaseTable for RemoteTable { let (request_id, response) = self.client.send(request, true).await?; self.check_table_response(&request_id, response).await?; + self.checkout_latest().await?; Ok(()) }