Merge pull request #1042 from benjaminsavage/walr2

Walr2
private-attribution · May 11, 2024 · eedf8d7 · eedf8d7
2 parents 3ec4f9c + d0fc05f
commit eedf8d7
Show file tree

Hide file tree

Showing 5 changed files with 171 additions and 108 deletions.
diff --git a/ipa-core/src/protocol/basics/reveal.rs b/ipa-core/src/protocol/basics/reveal.rs
@@ -223,9 +223,12 @@ mod tests {
         },
         rand::{thread_rng, Rng},
         secret_sharing::{
-            replicated::malicious::{
-                AdditiveShare as MaliciousReplicated, ExtendableField,
-                ThisCodeIsAuthorizedToDowngradeFromMalicious,
+            replicated::{
+                malicious::{
+                    AdditiveShare as MaliciousReplicated, ExtendableField,
+                    ThisCodeIsAuthorizedToDowngradeFromMalicious,
+                },
+                semi_honest::AdditiveShare,
             },
             IntoShares, SharedValue,
         },
@@ -299,12 +302,15 @@ mod tests {
 
         let input = rng.gen::<TestField>();
         let results = world
-            .semi_honest(input, |ctx, share| async move {
-                share
-                    .reveal(ctx.set_total_records(1), RecordId::from(0))
-                    .await
-                    .unwrap()
-            })
+            .semi_honest(
+                input,
+                |ctx, share: AdditiveShare<Fp32BitPrime, 32>| async move {
+                    share
+                        .reveal(ctx.set_total_records(1), RecordId::from(0))
+                        .await
+                        .unwrap()
+                },
+            )
             .await;
 
         assert_eq!(input, results[0]);