ml-kem: use module_lattice::encode

ml-kem/Cargo.toml

@@ -25,7 +25,7 @@ pkcs8 = ["dep:const-oid", "dep:pkcs8"]
 zeroize = ["module-lattice/zeroize", "dep:zeroize"]
 
 [dependencies]
-array = { package = "hybrid-array", version = "0.4.4", features = ["extra-sizes", "subtle"] }
+array = { version = "0.4.4", package = "hybrid-array", features = ["extra-sizes", "subtle"] }
 module-lattice = { version = "0.1.0-pre.0", features = ["subtle"] }
 kem = "0.3.0-rc.2"
 rand_core = "0.10.0-rc-6"

ml-kem/src/algebra.rs

@@ -1,13 +1,13 @@
 use array::{Array, typenum::U256};
 use module_lattice::{
     algebra::{Field, MultiplyNtt},
+    encode::Encode,
     util::Truncate,
 };
 use sha3::digest::XofReader;
 
 use crate::B32;
 use crate::crypto::{PRF, PrfOutput, XOF};
-use crate::encode::Encode;
 use crate::param::{ArraySize, CbdSamplingSize};
 
 module_lattice::define_field!(BaseField, u16, u32, u64, 3329);

ml-kem/src/lib.rs

@@ -57,9 +57,6 @@ mod crypto;
 /// Section 4.2.1. Conversion and Compression Algorithms, Compression and decompression
 mod compress;
 
-/// Section 4.2.1. Conversion and Compression Algorithms, Encoding and decoding
-mod encode;
-
 /// Section 5. The K-PKE Component Scheme
 mod pke;
 
@@ -78,7 +75,8 @@ pub use array;
 pub use ml_kem_512::MlKem512Params;
 pub use ml_kem_768::MlKem768Params;
 pub use ml_kem_1024::MlKem1024Params;
-pub use param::{ArraySize, ExpandedDecapsulationKey, ParameterSet};
+pub use module_lattice::encode::ArraySize;
+pub use param::{ExpandedDecapsulationKey, ParameterSet};
 pub use traits::*;
 
 use array::{