chore: rename kmeans_dimension_reduction to kmeans_dimension (#376)

Signed-off-by: usamoi <usamoi@outlook.com>

Author: usamoi

src/index/vchordrq/am/am_build.rs

@@ -1275,12 +1275,12 @@ fn make_internal_build(
 ) -> Vec<Structure<Normalized>> {
     use humansize::{BINARY, format_size};
     use std::iter::once;
-    let (reduction, sample_dim) = match internal_build.kmeans_dimension_reduction {
+    let (reduction, sample_dim) = match internal_build.kmeans_dimension {
         None => (None, vector_options.dims as usize),
         Some(d) if d < vector_options.dims => (Some(d as usize), d as usize),
         Some(d) => {
             pgrx::warning!(
-                "ignoring `kmeans_dimension_reduction = {}` because it is less than the vector dimension {}",
+                "ignoring `kmeans_dimension = {}` because it is less than the vector dimension {}",
                 d,
                 vector_options.dims
             );

src/index/vchordrq/types.rs

@@ -54,9 +54,9 @@ pub struct VchordrqInternalBuildOptions {
     pub build_threads: u16,
     #[serde(default = "VchordrqInternalBuildOptions::default_kmeans_algorithm")]
     pub kmeans_algorithm: KMeansAlgorithm,
-    #[serde(default = "VchordrqInternalBuildOptions::default_kmeans_dimension_reduction")]
+    #[serde(default = "VchordrqInternalBuildOptions::default_kmeans_dimension")]
     #[validate(range(min = 1, max = 16000))]
-    pub kmeans_dimension_reduction: Option<u32>,
+    pub kmeans_dimension: Option<u32>,
 }
 
 impl VchordrqInternalBuildOptions {
@@ -87,7 +87,7 @@ impl VchordrqInternalBuildOptions {
     fn default_kmeans_algorithm() -> KMeansAlgorithm {
         KMeansAlgorithm::Lloyd {}
     }
-    fn default_kmeans_dimension_reduction() -> Option<u32> {
+    fn default_kmeans_dimension() -> Option<u32> {
         None
     }
 }
@@ -101,7 +101,7 @@ impl Default for VchordrqInternalBuildOptions {
             kmeans_iterations: Self::default_kmeans_iterations(),
             build_threads: Self::default_build_threads(),
             kmeans_algorithm: Self::default_kmeans_algorithm(),
-            kmeans_dimension_reduction: Self::default_kmeans_dimension_reduction(),
+            kmeans_dimension: Self::default_kmeans_dimension(),
         }
     }
 }

tests/vchordrq/approximate.slt

@@ -11,7 +11,7 @@ residual_quantization = true
 [build.internal]
 lists = [33]
 spherical_centroids = false
-kmeans_dimension_reduction = 2
+kmeans_dimension = 2
 kmeans_algorithm.hierarchical = {}
 $$);