From a98865a20e3a33da20c9e7c50f7e9c66be93f57b Mon Sep 17 00:00:00 2001
From: xvzcf <xvzcf@users.noreply.github.com>
Date: Fri, 24 May 2024 10:51:58 -0400
Subject: [PATCH] Another rename.

---
 libcrux-ml-dsa/src/matrix.rs | 4 ++--
 libcrux-ml-dsa/src/sample.rs | 6 +++---
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/libcrux-ml-dsa/src/matrix.rs b/libcrux-ml-dsa/src/matrix.rs
index 6196b4ead..f7d5ede1e 100644
--- a/libcrux-ml-dsa/src/matrix.rs
+++ b/libcrux-ml-dsa/src/matrix.rs
@@ -1,4 +1,4 @@
-use crate::{arithmetic::PolynomialRingElement, sample::sample_ring_element_for_A};
+use crate::{arithmetic::PolynomialRingElement, sample::sample_ring_element_uniform};
 
 #[allow(non_snake_case)]
 #[inline(always)]
@@ -13,7 +13,7 @@ pub(crate) fn expand_to_A<const ROWS_IN_A: usize, const COLUMNS_IN_A: usize>(
             seed[32] = i as u8;
             seed[33] = j as u8;
 
-            let sampled = sample_ring_element_for_A(seed);
+            let sampled = sample_ring_element_uniform(seed);
 
             if transposed {
                 A[j][i] = sampled;
diff --git a/libcrux-ml-dsa/src/sample.rs b/libcrux-ml-dsa/src/sample.rs
index 143e70dbe..05fd2c9f9 100644
--- a/libcrux-ml-dsa/src/sample.rs
+++ b/libcrux-ml-dsa/src/sample.rs
@@ -27,7 +27,7 @@ fn sample_from_uniform_distribution_next(
 }
 
 #[allow(non_snake_case)]
-pub(crate) fn sample_ring_element_for_A(seed: [u8; 34]) -> PolynomialRingElement {
+pub(crate) fn sample_ring_element_uniform(seed: [u8; 34]) -> PolynomialRingElement {
     let mut state = XOF::new(seed);
     let randomness = XOF::squeeze_first_five_blocks(&mut state);
 
@@ -51,7 +51,7 @@ mod tests {
 
     #[allow(non_snake_case)]
     #[test]
-    fn test_sample_ring_element_for_A() {
+    fn test_sample_ring_element_uniform() {
         let seed: [u8; 34] = [
             33, 192, 250, 216, 117, 61, 16, 12, 248, 51, 213, 110, 64, 57, 119, 80, 164, 83, 73,
             91, 80, 128, 195, 219, 203, 149, 170, 233, 16, 232, 209, 105, 4, 5,
@@ -88,7 +88,7 @@ mod tests {
         ];
 
         assert_eq!(
-            sample_ring_element_for_A(seed).coefficients,
+            sample_ring_element_uniform(seed).coefficients,
             expected_coefficients
         );
     }