Compare commits

...

6 Commits

Author SHA1 Message Date
Wyatt Alt
74a5ab9e9a update cargo.lock 2025-08-29 09:25:05 -07:00
Wyatt Alt
59102d723a Update lance again 2025-08-29 08:39:36 -07:00
Wyatt Alt
e32dee9f78 lint fix 2025-08-28 16:44:18 -07:00
Wyatt Alt
af226f087a fix test issue 2025-08-28 16:27:27 -07:00
Wyatt Alt
521c3af6f0 update lance 2025-08-28 16:27:06 -07:00
Wyatt Alt
140a4e2f53 chore: add storage_options to object_store wrap()
Adds storage_options to object_store wrap() to adhere to upstream lance
change.
2025-08-28 16:27:06 -07:00
7 changed files with 138 additions and 52 deletions

142
Cargo.lock generated
View File

@@ -1399,7 +1399,7 @@ version = "0.6.0"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "6fac387a98bb7c37292057cffc56d62ecb629900026402633ae9160df93a8766" checksum = "6fac387a98bb7c37292057cffc56d62ecb629900026402633ae9160df93a8766"
dependencies = [ dependencies = [
"nom", "nom 7.1.3",
] ]
[[package]] [[package]]
@@ -2717,7 +2717,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "74fef4569247a5f429d9156b9d0a2599914385dd189c539334c625d8099d90ab" checksum = "74fef4569247a5f429d9156b9d0a2599914385dd189c539334c625d8099d90ab"
dependencies = [ dependencies = [
"futures-core", "futures-core",
"nom", "nom 7.1.3",
"pin-project-lite", "pin-project-lite",
] ]
@@ -2744,6 +2744,12 @@ version = "0.2.0"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "95765f67b4b18863968b4a1bd5bb576f732b29a4a28c7cd84c09fa3e2875f33c" checksum = "95765f67b4b18863968b4a1bd5bb576f732b29a4a28c7cd84c09fa3e2875f33c"
[[package]]
name = "fast-float2"
version = "0.2.3"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "f8eb564c5c7423d25c886fb561d1e4ee69f72354d16918afa32c08811f6b6a55"
[[package]] [[package]]
name = "fastdivide" name = "fastdivide"
version = "0.4.2" version = "0.4.2"
@@ -2837,8 +2843,8 @@ checksum = "42703706b716c37f96a77aea830392ad231f44c9e9a67872fa5548707e11b11c"
[[package]] [[package]]
name = "fsst" name = "fsst"
version = "0.33.0" version = "0.34.0"
source = "git+https://github.com/lancedb/lance.git?tag=v0.33.0-beta.4#e37e9df2458e88c37205415bd00d950b7f936061" source = "git+https://github.com/lancedb/lance.git?tag=v0.34.0-beta.2#0b82c816d1b3cf5a60a42b7cfa525c2df971aa7f"
dependencies = [ dependencies = [
"arrow-array", "arrow-array",
"rand 0.9.1", "rand 0.9.1",
@@ -3890,10 +3896,12 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "be1f93b8b1eb69c77f24bbb0afdf66f54b632ee39af40ca21c4365a1d7347e49" checksum = "be1f93b8b1eb69c77f24bbb0afdf66f54b632ee39af40ca21c4365a1d7347e49"
dependencies = [ dependencies = [
"jiff-static", "jiff-static",
"jiff-tzdb-platform",
"log", "log",
"portable-atomic", "portable-atomic",
"portable-atomic-util", "portable-atomic-util",
"serde", "serde",
"windows-sys 0.52.0",
] ]
[[package]] [[package]]
@@ -3907,6 +3915,21 @@ dependencies = [
"syn 2.0.103", "syn 2.0.103",
] ]
[[package]]
name = "jiff-tzdb"
version = "0.1.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "c1283705eb0a21404d2bfd6eef2a7593d240bc42a0bdb39db0ad6fa2ec026524"
[[package]]
name = "jiff-tzdb-platform"
version = "0.1.3"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "875a5a69ac2bab1a891711cf5eccbec1ce0341ea805560dcd90b7a2e925132e8"
dependencies = [
"jiff-tzdb",
]
[[package]] [[package]]
name = "jni" name = "jni"
version = "0.21.1" version = "0.21.1"
@@ -3949,10 +3972,30 @@ dependencies = [
"wasm-bindgen", "wasm-bindgen",
] ]
[[package]]
name = "jsonb"
version = "0.5.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "a452366d21e8d3cbca680c41388e01d6a88739afef7877961946a6da409f9ccd"
dependencies = [
"byteorder",
"ethnum",
"fast-float2",
"itoa",
"jiff",
"nom 8.0.0",
"num-traits",
"ordered-float",
"rand 0.9.1",
"ryu",
"serde",
"serde_json",
]
[[package]] [[package]]
name = "lance" name = "lance"
version = "0.33.0" version = "0.34.0"
source = "git+https://github.com/lancedb/lance.git?tag=v0.33.0-beta.4#e37e9df2458e88c37205415bd00d950b7f936061" source = "git+https://github.com/lancedb/lance.git?tag=v0.34.0-beta.2#0b82c816d1b3cf5a60a42b7cfa525c2df971aa7f"
dependencies = [ dependencies = [
"arrow", "arrow",
"arrow-arith", "arrow-arith",
@@ -4007,6 +4050,7 @@ dependencies = [
"tantivy", "tantivy",
"tempfile", "tempfile",
"tokio", "tokio",
"tokio-stream",
"tracing", "tracing",
"url", "url",
"uuid", "uuid",
@@ -4014,8 +4058,8 @@ dependencies = [
[[package]] [[package]]
name = "lance-arrow" name = "lance-arrow"
version = "0.33.0" version = "0.34.0"
source = "git+https://github.com/lancedb/lance.git?tag=v0.33.0-beta.4#e37e9df2458e88c37205415bd00d950b7f936061" source = "git+https://github.com/lancedb/lance.git?tag=v0.34.0-beta.2#0b82c816d1b3cf5a60a42b7cfa525c2df971aa7f"
dependencies = [ dependencies = [
"arrow-array", "arrow-array",
"arrow-buffer", "arrow-buffer",
@@ -4026,14 +4070,25 @@ dependencies = [
"bytes", "bytes",
"getrandom 0.2.16", "getrandom 0.2.16",
"half", "half",
"jsonb",
"num-traits", "num-traits",
"rand 0.9.1", "rand 0.9.1",
] ]
[[package]]
name = "lance-bitpacking"
version = "0.34.0"
source = "git+https://github.com/lancedb/lance.git?tag=v0.34.0-beta.2#0b82c816d1b3cf5a60a42b7cfa525c2df971aa7f"
dependencies = [
"arrayref",
"paste",
"seq-macro",
]
[[package]] [[package]]
name = "lance-core" name = "lance-core"
version = "0.33.0" version = "0.34.0"
source = "git+https://github.com/lancedb/lance.git?tag=v0.33.0-beta.4#e37e9df2458e88c37205415bd00d950b7f936061" source = "git+https://github.com/lancedb/lance.git?tag=v0.34.0-beta.2#0b82c816d1b3cf5a60a42b7cfa525c2df971aa7f"
dependencies = [ dependencies = [
"arrow-array", "arrow-array",
"arrow-buffer", "arrow-buffer",
@@ -4068,8 +4123,8 @@ dependencies = [
[[package]] [[package]]
name = "lance-datafusion" name = "lance-datafusion"
version = "0.33.0" version = "0.34.0"
source = "git+https://github.com/lancedb/lance.git?tag=v0.33.0-beta.4#e37e9df2458e88c37205415bd00d950b7f936061" source = "git+https://github.com/lancedb/lance.git?tag=v0.34.0-beta.2#0b82c816d1b3cf5a60a42b7cfa525c2df971aa7f"
dependencies = [ dependencies = [
"arrow", "arrow",
"arrow-array", "arrow-array",
@@ -4083,6 +4138,7 @@ dependencies = [
"datafusion-functions", "datafusion-functions",
"datafusion-physical-expr", "datafusion-physical-expr",
"futures", "futures",
"jsonb",
"lance-arrow", "lance-arrow",
"lance-core", "lance-core",
"lance-datagen", "lance-datagen",
@@ -4097,8 +4153,8 @@ dependencies = [
[[package]] [[package]]
name = "lance-datagen" name = "lance-datagen"
version = "0.33.0" version = "0.34.0"
source = "git+https://github.com/lancedb/lance.git?tag=v0.33.0-beta.4#e37e9df2458e88c37205415bd00d950b7f936061" source = "git+https://github.com/lancedb/lance.git?tag=v0.34.0-beta.2#0b82c816d1b3cf5a60a42b7cfa525c2df971aa7f"
dependencies = [ dependencies = [
"arrow", "arrow",
"arrow-array", "arrow-array",
@@ -4115,11 +4171,9 @@ dependencies = [
[[package]] [[package]]
name = "lance-encoding" name = "lance-encoding"
version = "0.33.0" version = "0.34.0"
source = "git+https://github.com/lancedb/lance.git?tag=v0.33.0-beta.4#e37e9df2458e88c37205415bd00d950b7f936061" source = "git+https://github.com/lancedb/lance.git?tag=v0.34.0-beta.2#0b82c816d1b3cf5a60a42b7cfa525c2df971aa7f"
dependencies = [ dependencies = [
"arrayref",
"arrow",
"arrow-arith", "arrow-arith",
"arrow-array", "arrow-array",
"arrow-buffer", "arrow-buffer",
@@ -4136,16 +4190,15 @@ dependencies = [
"hyperloglogplus", "hyperloglogplus",
"itertools 0.13.0", "itertools 0.13.0",
"lance-arrow", "lance-arrow",
"lance-bitpacking",
"lance-core", "lance-core",
"log", "log",
"lz4", "lz4",
"num-traits", "num-traits",
"paste",
"prost", "prost",
"prost-build", "prost-build",
"prost-types", "prost-types",
"rand 0.9.1", "rand 0.9.1",
"seq-macro",
"snafu", "snafu",
"tokio", "tokio",
"tracing", "tracing",
@@ -4155,8 +4208,8 @@ dependencies = [
[[package]] [[package]]
name = "lance-file" name = "lance-file"
version = "0.33.0" version = "0.34.0"
source = "git+https://github.com/lancedb/lance.git?tag=v0.33.0-beta.4#e37e9df2458e88c37205415bd00d950b7f936061" source = "git+https://github.com/lancedb/lance.git?tag=v0.34.0-beta.2#0b82c816d1b3cf5a60a42b7cfa525c2df971aa7f"
dependencies = [ dependencies = [
"arrow-arith", "arrow-arith",
"arrow-array", "arrow-array",
@@ -4190,8 +4243,8 @@ dependencies = [
[[package]] [[package]]
name = "lance-index" name = "lance-index"
version = "0.33.0" version = "0.34.0"
source = "git+https://github.com/lancedb/lance.git?tag=v0.33.0-beta.4#e37e9df2458e88c37205415bd00d950b7f936061" source = "git+https://github.com/lancedb/lance.git?tag=v0.34.0-beta.2#0b82c816d1b3cf5a60a42b7cfa525c2df971aa7f"
dependencies = [ dependencies = [
"arrow", "arrow",
"arrow-array", "arrow-array",
@@ -4245,8 +4298,8 @@ dependencies = [
[[package]] [[package]]
name = "lance-io" name = "lance-io"
version = "0.33.0" version = "0.34.0"
source = "git+https://github.com/lancedb/lance.git?tag=v0.33.0-beta.4#e37e9df2458e88c37205415bd00d950b7f936061" source = "git+https://github.com/lancedb/lance.git?tag=v0.34.0-beta.2#0b82c816d1b3cf5a60a42b7cfa525c2df971aa7f"
dependencies = [ dependencies = [
"arrow", "arrow",
"arrow-arith", "arrow-arith",
@@ -4286,8 +4339,8 @@ dependencies = [
[[package]] [[package]]
name = "lance-linalg" name = "lance-linalg"
version = "0.33.0" version = "0.34.0"
source = "git+https://github.com/lancedb/lance.git?tag=v0.33.0-beta.4#e37e9df2458e88c37205415bd00d950b7f936061" source = "git+https://github.com/lancedb/lance.git?tag=v0.34.0-beta.2#0b82c816d1b3cf5a60a42b7cfa525c2df971aa7f"
dependencies = [ dependencies = [
"arrow-array", "arrow-array",
"arrow-buffer", "arrow-buffer",
@@ -4310,8 +4363,8 @@ dependencies = [
[[package]] [[package]]
name = "lance-table" name = "lance-table"
version = "0.33.0" version = "0.34.0"
source = "git+https://github.com/lancedb/lance.git?tag=v0.33.0-beta.4#e37e9df2458e88c37205415bd00d950b7f936061" source = "git+https://github.com/lancedb/lance.git?tag=v0.34.0-beta.2#0b82c816d1b3cf5a60a42b7cfa525c2df971aa7f"
dependencies = [ dependencies = [
"arrow", "arrow",
"arrow-array", "arrow-array",
@@ -4349,8 +4402,8 @@ dependencies = [
[[package]] [[package]]
name = "lance-testing" name = "lance-testing"
version = "0.33.0" version = "0.34.0"
source = "git+https://github.com/lancedb/lance.git?tag=v0.33.0-beta.4#e37e9df2458e88c37205415bd00d950b7f936061" source = "git+https://github.com/lancedb/lance.git?tag=v0.34.0-beta.2#0b82c816d1b3cf5a60a42b7cfa525c2df971aa7f"
dependencies = [ dependencies = [
"arrow-array", "arrow-array",
"arrow-schema", "arrow-schema",
@@ -4970,6 +5023,15 @@ dependencies = [
"minimal-lexical", "minimal-lexical",
] ]
[[package]]
name = "nom"
version = "8.0.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "df9761775871bdef83bee530e60050f7e54b1105350d6884eb0fb4f46c2f9405"
dependencies = [
"memchr",
]
[[package]] [[package]]
name = "now" name = "now"
version = "0.1.3" version = "0.1.3"
@@ -5257,6 +5319,15 @@ version = "0.2.0"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "04744f49eae99ab78e0d5c0b603ab218f515ea8cfe5a456d7629ad883a3b6e7d" checksum = "04744f49eae99ab78e0d5c0b603ab218f515ea8cfe5a456d7629ad883a3b6e7d"
[[package]]
name = "ordered-float"
version = "5.0.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "e2c1f9f56e534ac6a9b8a4600bdf0f530fb393b5f393e7b4d03489c3cf0c3f01"
dependencies = [
"num-traits",
]
[[package]] [[package]]
name = "outref" name = "outref"
version = "0.5.2" version = "0.5.2"
@@ -6534,7 +6605,7 @@ dependencies = [
"futures-core", "futures-core",
"futures-timer", "futures-timer",
"mime", "mime",
"nom", "nom 7.1.3",
"pin-project-lite", "pin-project-lite",
"reqwest", "reqwest",
"thiserror 1.0.69", "thiserror 1.0.69",
@@ -6942,6 +7013,7 @@ version = "1.0.140"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "20068b6e96dc6c9bd23e01df8827e6c7e1f2fddd43c21810382803c136b99373" checksum = "20068b6e96dc6c9bd23e01df8827e6c7e1f2fddd43c21810382803c136b99373"
dependencies = [ dependencies = [
"indexmap 2.9.0",
"itoa", "itoa",
"memchr", "memchr",
"ryu", "ryu",
@@ -7168,7 +7240,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "5851699c4033c63636f7ea4cf7b7c1f1bf06d0cc03cfb42e711de5a5c46cf326" checksum = "5851699c4033c63636f7ea4cf7b7c1f1bf06d0cc03cfb42e711de5a5c46cf326"
dependencies = [ dependencies = [
"base64 0.13.1", "base64 0.13.1",
"nom", "nom 7.1.3",
"serde", "serde",
"unicode-segmentation", "unicode-segmentation",
] ]
@@ -7505,7 +7577,7 @@ version = "0.24.0"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "e810cdeeebca57fc3f7bfec5f85fdbea9031b2ac9b990eb5ff49b371d52bbe6a" checksum = "e810cdeeebca57fc3f7bfec5f85fdbea9031b2ac9b990eb5ff49b371d52bbe6a"
dependencies = [ dependencies = [
"nom", "nom 7.1.3",
"serde", "serde",
"serde_json", "serde_json",
] ]

View File

@@ -15,14 +15,14 @@ categories = ["database-implementations"]
rust-version = "1.78.0" rust-version = "1.78.0"
[workspace.dependencies] [workspace.dependencies]
lance = { "version" = "=0.33.0", default-features = false, "features" = ["dynamodb"], "tag" = "v0.33.0-beta.4", "git" = "https://github.com/lancedb/lance.git" } lance = { "version" = "=0.34.0", default-features = false, "features" = ["dynamodb"], "tag" = "v0.34.0-beta.2", "git" = "https://github.com/lancedb/lance.git" }
lance-io = { "version" = "=0.33.0", default-features = false, "tag" = "v0.33.0-beta.4", "git" = "https://github.com/lancedb/lance.git" } lance-io = { "version" = "=0.34.0", default-features = false, "tag" = "v0.34.0-beta.2", "git" = "https://github.com/lancedb/lance.git" }
lance-index = { "version" = "=0.33.0", "tag" = "v0.33.0-beta.4", "git" = "https://github.com/lancedb/lance.git" } lance-index = { "version" = "=0.34.0", "tag" = "v0.34.0-beta.2", "git" = "https://github.com/lancedb/lance.git" }
lance-linalg = { "version" = "=0.33.0", "tag" = "v0.33.0-beta.4", "git" = "https://github.com/lancedb/lance.git" } lance-linalg = { "version" = "=0.34.0", "tag" = "v0.34.0-beta.2", "git" = "https://github.com/lancedb/lance.git" }
lance-table = { "version" = "=0.33.0", "tag" = "v0.33.0-beta.4", "git" = "https://github.com/lancedb/lance.git" } lance-table = { "version" = "=0.34.0", "tag" = "v0.34.0-beta.2", "git" = "https://github.com/lancedb/lance.git" }
lance-testing = { "version" = "=0.33.0", "tag" = "v0.33.0-beta.4", "git" = "https://github.com/lancedb/lance.git" } lance-testing = { "version" = "=0.34.0", "tag" = "v0.34.0-beta.2", "git" = "https://github.com/lancedb/lance.git" }
lance-datafusion = { "version" = "=0.33.0", "tag" = "v0.33.0-beta.4", "git" = "https://github.com/lancedb/lance.git" } lance-datafusion = { "version" = "=0.34.0", "tag" = "v0.34.0-beta.2", "git" = "https://github.com/lancedb/lance.git" }
lance-encoding = { "version" = "=0.33.0", "tag" = "v0.33.0-beta.4", "git" = "https://github.com/lancedb/lance.git" } lance-encoding = { "version" = "=0.34.0", "tag" = "v0.34.0-beta.2", "git" = "https://github.com/lancedb/lance.git" }
# Note that this one does not include pyarrow # Note that this one does not include pyarrow
arrow = { version = "55.1", optional = false } arrow = { version = "55.1", optional = false }
arrow-array = "55.1" arrow-array = "55.1"

View File

@@ -94,7 +94,7 @@ impl napi::bindgen_prelude::FromNapiValue for Session {
env: napi::sys::napi_env, env: napi::sys::napi_env,
napi_val: napi::sys::napi_value, napi_val: napi::sys::napi_value,
) -> napi::Result<Self> { ) -> napi::Result<Self> {
let object: napi::bindgen_prelude::ClassInstance<Session> = let object: napi::bindgen_prelude::ClassInstance<Self> =
napi::bindgen_prelude::ClassInstance::from_napi_value(env, napi_val)?; napi::bindgen_prelude::ClassInstance::from_napi_value(env, napi_val)?;
let copy = object.clone(); let copy = object.clone();
Ok(copy) Ok(copy)

View File

@@ -170,7 +170,11 @@ impl MirroringObjectStoreWrapper {
} }
impl WrappingObjectStore for MirroringObjectStoreWrapper { impl WrappingObjectStore for MirroringObjectStoreWrapper {
fn wrap(&self, primary: Arc<dyn ObjectStore>) -> Arc<dyn ObjectStore> { fn wrap(
&self,
primary: Arc<dyn ObjectStore>,
_storage_options: Option<&std::collections::HashMap<String, String>>,
) -> Arc<dyn ObjectStore> {
Arc::new(MirroringObjectStore { Arc::new(MirroringObjectStore {
primary, primary,
secondary: self.secondary.clone(), secondary: self.secondary.clone(),

View File

@@ -50,7 +50,11 @@ impl IoStatsHolder {
} }
impl WrappingObjectStore for IoStatsHolder { impl WrappingObjectStore for IoStatsHolder {
fn wrap(&self, target: Arc<dyn ObjectStore>) -> Arc<dyn ObjectStore> { fn wrap(
&self,
target: Arc<dyn ObjectStore>,
_storage_options: Option<&std::collections::HashMap<String, String>>,
) -> Arc<dyn ObjectStore> {
Arc::new(IoTrackingStore { Arc::new(IoTrackingStore {
target, target,
stats: self.0.clone(), stats: self.0.clone(),

View File

@@ -2147,7 +2147,7 @@ mod tests {
let body = request.body().unwrap().as_bytes().unwrap(); let body = request.body().unwrap().as_bytes().unwrap();
let body: serde_json::Value = serde_json::from_slice(body).unwrap(); let body: serde_json::Value = serde_json::from_slice(body).unwrap();
let mut expected_body = serde_json::json!({ let expected_body = serde_json::json!({
"prefilter": true, "prefilter": true,
"distance_type": "l2", "distance_type": "l2",
"nprobes": 20, "nprobes": 20,
@@ -2156,12 +2156,11 @@ mod tests {
"lower_bound": Option::<f32>::None, "lower_bound": Option::<f32>::None,
"upper_bound": Option::<f32>::None, "upper_bound": Option::<f32>::None,
"k": 10, "k": 10,
"vector": vector_f32_to_f64(&[0.1, 0.2, 0.3]),
"ef": Option::<usize>::None, "ef": Option::<usize>::None,
"refine_factor": null, "refine_factor": null,
"version": null, "version": null,
}); });
// Pass vector separately to make sure it matches f32 precision.
expected_body["vector"] = vec![0.1f32, 0.2, 0.3].into();
assert_eq!(body, expected_body); assert_eq!(body, expected_body);
let response_body = write_ipc_file(&expected_data_ref); let response_body = write_ipc_file(&expected_data_ref);
@@ -2232,6 +2231,13 @@ mod tests {
assert_eq!(data[0].as_ref().unwrap(), &expected_data); assert_eq!(data[0].as_ref().unwrap(), &expected_data);
} }
fn vector_f32_to_f64(vector: &[f32]) -> Vec<serde_json::Value> {
vector
.iter()
.map(|v| serde_json::Value::Number(serde_json::Number::from_f64(*v as f64).unwrap()))
.collect()
}
#[tokio::test] #[tokio::test]
async fn test_query_vector_all_params() { async fn test_query_vector_all_params() {
let table = Table::new_with_handler("my_table", |request| { let table = Table::new_with_handler("my_table", |request| {
@@ -2244,7 +2250,7 @@ mod tests {
let body = request.body().unwrap().as_bytes().unwrap(); let body = request.body().unwrap().as_bytes().unwrap();
let body: serde_json::Value = serde_json::from_slice(body).unwrap(); let body: serde_json::Value = serde_json::from_slice(body).unwrap();
let mut expected_body = serde_json::json!({ let expected_body = serde_json::json!({
"vector_column": "my_vector", "vector_column": "my_vector",
"prefilter": false, "prefilter": false,
"k": 42, "k": 42,
@@ -2255,14 +2261,13 @@ mod tests {
"nprobes": 12, "nprobes": 12,
"minimum_nprobes": 12, "minimum_nprobes": 12,
"maximum_nprobes": 12, "maximum_nprobes": 12,
"vector": vector_f32_to_f64(&[0.1, 0.2, 0.3]),
"lower_bound": Option::<f32>::None, "lower_bound": Option::<f32>::None,
"upper_bound": Option::<f32>::None, "upper_bound": Option::<f32>::None,
"ef": Option::<usize>::None, "ef": Option::<usize>::None,
"refine_factor": 2, "refine_factor": 2,
"version": null, "version": null,
}); });
// Pass vector separately to make sure it matches f32 precision.
expected_body["vector"] = vec![0.1f32, 0.2, 0.3].into();
assert_eq!(body, expected_body); assert_eq!(body, expected_body);
let data = RecordBatch::try_new( let data = RecordBatch::try_new(

View File

@@ -3264,6 +3264,7 @@ mod tests {
fn wrap( fn wrap(
&self, &self,
original: Arc<dyn object_store::ObjectStore>, original: Arc<dyn object_store::ObjectStore>,
_storage_options: Option<&std::collections::HashMap<String, String>>,
) -> Arc<dyn object_store::ObjectStore> { ) -> Arc<dyn object_store::ObjectStore> {
self.called.store(true, Ordering::Relaxed); self.called.store(true, Ordering::Relaxed);
original original