Skip to content

Commit ee1141a

Browse files
committed
BinaryQuantizationQueryEncoding variants
1 parent 97205c4 commit ee1141a

File tree

1 file changed

+26
-2
lines changed

1 file changed

+26
-2
lines changed

src/builders/binary_quantization_builder.rs

Lines changed: 26 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -117,10 +117,34 @@ impl From<String> for BinaryQuantizationBuilderError {
117117
}
118118

119119
impl BinaryQuantizationQueryEncoding {
120-
pub fn new_setting(setting: binary_quantization_query_encoding::Setting) -> Self {
120+
pub fn scalar8bits() -> Self {
121121
Self {
122122
variant: Some(binary_quantization_query_encoding::Variant::Setting(
123-
setting.into(),
123+
binary_quantization_query_encoding::Setting::Scalar8Bits.into(),
124+
)),
125+
}
126+
}
127+
128+
pub fn scalar4bits() -> Self {
129+
Self {
130+
variant: Some(binary_quantization_query_encoding::Variant::Setting(
131+
binary_quantization_query_encoding::Setting::Scalar4Bits.into(),
132+
)),
133+
}
134+
}
135+
136+
pub fn binary() -> Self {
137+
Self {
138+
variant: Some(binary_quantization_query_encoding::Variant::Setting(
139+
binary_quantization_query_encoding::Setting::Binary.into(),
140+
)),
141+
}
142+
}
143+
144+
pub fn default_encoding() -> Self {
145+
Self {
146+
variant: Some(binary_quantization_query_encoding::Variant::Setting(
147+
binary_quantization_query_encoding::Setting::Default.into(),
124148
)),
125149
}
126150
}

0 commit comments

Comments
 (0)