Small change to add curta constraints in mock_build

succinctlabs · Sep 19, 2023 · 3caffda · 3caffda
1 parent 760823a
commit 3caffda
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/plonky2x/src/frontend/builder/mod.rs b/plonky2x/src/frontend/builder/mod.rs
@@ -151,7 +151,10 @@ impl<L: PlonkParameters<D>, const D: usize> CircuitBuilder<L, D> {
         CircuitBuild { data, io: self.io }
     }
 
-    pub fn mock_build(self) -> MockCircuitBuild<L, D> {
+    pub fn mock_build(mut self) -> MockCircuitBuild<L, D> {
+        if !self.sha256_requests.is_empty() {
+            self.curta_constrain_sha256();
+        }
         let mock_circuit = self.api.mock_build();
         MockCircuitBuild {
             data: mock_circuit,

diff --git a/plonky2x/src/frontend/num/nonnative/nonnative.rs b/plonky2x/src/frontend/num/nonnative/nonnative.rs
@@ -57,7 +57,7 @@ impl<FF: PrimeField> CircuitVariable for NonNativeTarget<FF> {
                 .constants
                 .insert(*variable, L::Field::from_canonical_u32(big_int_rep[i]));
         }
-        return target;
+        target
     }
 
     fn variables(&self) -> Vec<Variable> {