chore(ci): fix doctest by using parameters with enough precision

zama-ai · Feb 26, 2024 · fa9dbc0 · fa9dbc0
1 parent e2e75a3
commit fa9dbc0
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 12 deletions.
diff --git a/tfhe/src/integer/wopbs/mod.rs b/tfhe/src/integer/wopbs/mod.rs
@@ -425,12 +425,12 @@ impl WopbsKey {
     /// # Example
     /// ```rust
     /// use tfhe::integer::gen_keys;
-    /// use tfhe::integer::parameters::PARAM_4_BITS_5_BLOCKS;
     /// use tfhe::integer::wopbs::WopbsKey;
+    /// use tfhe::shortint::parameters::parameters_wopbs_message_carry::WOPBS_PARAM_MESSAGE_4_CARRY_4_KS_PBS;
     ///
     /// let basis: Vec<u64> = vec![9, 11];
     ///
-    /// let param = PARAM_4_BITS_5_BLOCKS;
+    /// let param = WOPBS_PARAM_MESSAGE_4_CARRY_4_KS_PBS;
     /// //Generate the client key and the server key:
     /// let (cks, sks) = gen_keys(param);
     /// let wopbs_key = WopbsKey::new_wopbs_key_only_for_wopbs(&cks, &sks);
@@ -629,12 +629,12 @@ impl WopbsKey {
     ///
     /// ```rust
     /// use tfhe::integer::gen_keys;
-    /// use tfhe::integer::parameters::PARAM_4_BITS_5_BLOCKS;
     /// use tfhe::integer::wopbs::WopbsKey;
+    /// use tfhe::shortint::parameters::parameters_wopbs_message_carry::WOPBS_PARAM_MESSAGE_4_CARRY_4_KS_PBS;
     ///
     /// let basis: Vec<u64> = vec![9, 11];
     ///
-    /// let param = PARAM_4_BITS_5_BLOCKS;
+    /// let param = WOPBS_PARAM_MESSAGE_4_CARRY_4_KS_PBS;
     /// //Generate the client key and the server key:
     /// let (cks, sks) = gen_keys(param);
     /// let wopbs_key = WopbsKey::new_wopbs_key_only_for_wopbs(&cks, &sks);
@@ -941,12 +941,12 @@ impl WopbsKey {
     ///
     /// ```rust
     /// use tfhe::integer::gen_keys;
-    /// use tfhe::integer::parameters::PARAM_4_BITS_5_BLOCKS;
     /// use tfhe::integer::wopbs::WopbsKey;
+    /// use tfhe::shortint::parameters::parameters_wopbs_message_carry::WOPBS_PARAM_MESSAGE_4_CARRY_4_KS_PBS;
     ///
     /// let basis: Vec<u64> = vec![9, 11];
     ///
-    /// let param = PARAM_4_BITS_5_BLOCKS;
+    /// let param = WOPBS_PARAM_MESSAGE_4_CARRY_4_KS_PBS;
     /// //Generate the client key and the server key:
     /// let (cks, sks) = gen_keys(param);
     /// let wopbs_key = WopbsKey::new_wopbs_key_only_for_wopbs(&cks, &sks);
@@ -1013,12 +1013,12 @@ impl WopbsKey {
     ///
     /// ```rust
     /// use tfhe::integer::gen_keys;
-    /// use tfhe::integer::parameters::PARAM_4_BITS_5_BLOCKS;
     /// use tfhe::integer::wopbs::WopbsKey;
+    /// use tfhe::shortint::parameters::parameters_wopbs_message_carry::WOPBS_PARAM_MESSAGE_4_CARRY_4_KS_PBS;
     ///
     /// let basis: Vec<u64> = vec![9, 11];
     ///
-    /// let param = PARAM_4_BITS_5_BLOCKS;
+    /// let param = WOPBS_PARAM_MESSAGE_4_CARRY_4_KS_PBS;
     /// //Generate the client key and the server key:
     /// let (cks, sks) = gen_keys(param);
     /// let wopbs_key = WopbsKey::new_wopbs_key_only_for_wopbs(&cks, &sks);

diff --git a/tfhe/src/shortint/wopbs/mod.rs b/tfhe/src/shortint/wopbs/mod.rs
@@ -309,11 +309,11 @@ impl WopbsKey {
     /// ```rust
     /// use rand::Rng;
     /// use tfhe::shortint::gen_keys;
-    /// use tfhe::shortint::parameters::parameters_wopbs::WOPBS_PARAM_MESSAGE_3_NORM2_2_KS_PBS;
+    /// use tfhe::shortint::parameters::parameters_wopbs_message_carry::WOPBS_PARAM_MESSAGE_3_CARRY_3_KS_PBS;
     /// use tfhe::shortint::wopbs::WopbsKey;
     ///
     /// // Generate the client key and the server key:
-    /// let (cks, sks) = gen_keys(WOPBS_PARAM_MESSAGE_3_NORM2_2_KS_PBS);
+    /// let (cks, sks) = gen_keys(WOPBS_PARAM_MESSAGE_3_CARRY_3_KS_PBS);
     /// let wopbs_key = WopbsKey::new_wopbs_key_only_for_wopbs(&cks, &sks);
     /// let message_modulus = 5;
     /// let m = 2;
@@ -444,10 +444,10 @@ impl WopbsKey {
     ///
     /// ```rust
     /// use tfhe::shortint::gen_keys;
-    /// use tfhe::shortint::parameters::parameters_wopbs::WOPBS_PARAM_MESSAGE_3_NORM2_2_KS_PBS;
+    /// use tfhe::shortint::parameters::parameters_wopbs_message_carry::WOPBS_PARAM_MESSAGE_3_CARRY_3_KS_PBS;
     /// use tfhe::shortint::wopbs::*;
     ///
-    /// let (cks, sks) = gen_keys(WOPBS_PARAM_MESSAGE_3_NORM2_2_KS_PBS);
+    /// let (cks, sks) = gen_keys(WOPBS_PARAM_MESSAGE_3_CARRY_3_KS_PBS);
     /// let wopbs_key = WopbsKey::new_wopbs_key_only_for_wopbs(&cks, &sks);
     /// let msg = 2;
     /// let modulus = 5;