diff --git a/vortex-tensor/src/encodings/turboquant/array/mod.rs b/vortex-tensor/src/encodings/turboquant/array/mod.rs index 0c98c974203..e82313f1dc6 100644 --- a/vortex-tensor/src/encodings/turboquant/array/mod.rs +++ b/vortex-tensor/src/encodings/turboquant/array/mod.rs @@ -9,5 +9,3 @@ pub(crate) mod slots; pub(crate) mod centroids; pub(crate) mod rotation; - -pub(crate) mod scheme; diff --git a/vortex-tensor/src/encodings/turboquant/mod.rs b/vortex-tensor/src/encodings/turboquant/mod.rs index 3d8085b256e..18bc397bf9f 100644 --- a/vortex-tensor/src/encodings/turboquant/mod.rs +++ b/vortex-tensor/src/encodings/turboquant/mod.rs @@ -108,21 +108,20 @@ mod array; pub use array::data::TurboQuantArrayExt; pub use array::data::TurboQuantData; -pub use array::scheme::TurboQuantScheme; pub(crate) mod compute; mod metadata; mod vtable; + pub use vtable::TurboQuant; pub use vtable::TurboQuantArray; -mod compress; -pub use compress::TurboQuantConfig; -pub use compress::turboquant_encode; - -mod decompress; +mod scheme; +pub use scheme::TurboQuantScheme; +pub use scheme::compress::TurboQuantConfig; +pub use scheme::compress::turboquant_encode; #[cfg(test)] mod tests; diff --git a/vortex-tensor/src/encodings/turboquant/compress.rs b/vortex-tensor/src/encodings/turboquant/scheme/compress.rs similarity index 100% rename from vortex-tensor/src/encodings/turboquant/compress.rs rename to vortex-tensor/src/encodings/turboquant/scheme/compress.rs diff --git a/vortex-tensor/src/encodings/turboquant/decompress.rs b/vortex-tensor/src/encodings/turboquant/scheme/decompress.rs similarity index 100% rename from vortex-tensor/src/encodings/turboquant/decompress.rs rename to vortex-tensor/src/encodings/turboquant/scheme/decompress.rs diff --git a/vortex-tensor/src/encodings/turboquant/array/scheme.rs b/vortex-tensor/src/encodings/turboquant/scheme/mod.rs similarity index 98% rename from vortex-tensor/src/encodings/turboquant/array/scheme.rs rename to vortex-tensor/src/encodings/turboquant/scheme/mod.rs index 05ca8553073..d93afda0984 100644 --- a/vortex-tensor/src/encodings/turboquant/array/scheme.rs +++ b/vortex-tensor/src/encodings/turboquant/scheme/mod.rs @@ -1,7 +1,7 @@ // SPDX-License-Identifier: Apache-2.0 // SPDX-FileCopyrightText: Copyright the Vortex contributors -//! TurboQuant compression scheme for the pluggable compressor. +//! TurboQuant compression scheme and decompression. use vortex_array::ArrayRef; use vortex_array::Canonical; @@ -18,6 +18,9 @@ use crate::encodings::turboquant::TurboQuant; use crate::encodings::turboquant::TurboQuantConfig; use crate::encodings::turboquant::turboquant_encode; +pub(super) mod compress; +pub(super) mod decompress; + /// TurboQuant compression scheme for [`Vector`] extension types. /// /// Applies lossy vector quantization to [`Vector`] extension arrays using the TurboQuant diff --git a/vortex-tensor/src/encodings/turboquant/vtable.rs b/vortex-tensor/src/encodings/turboquant/vtable.rs index 9da8f242e7d..70002960a05 100644 --- a/vortex-tensor/src/encodings/turboquant/vtable.rs +++ b/vortex-tensor/src/encodings/turboquant/vtable.rs @@ -38,8 +38,8 @@ use crate::encodings::turboquant::TurboQuantData; use crate::encodings::turboquant::array::slots::Slot; use crate::encodings::turboquant::compute::rules::PARENT_KERNELS; use crate::encodings::turboquant::compute::rules::RULES; -use crate::encodings::turboquant::decompress::execute_decompress; use crate::encodings::turboquant::metadata::TurboQuantMetadata; +use crate::encodings::turboquant::scheme::decompress::execute_decompress; use crate::vector::AnyVector; use crate::vector::VectorMatcherMetadata;