Fix pil-helpers

0xPolygonHermez · Sep 9, 2024 · cac73ba · cac73ba
1 parent 99bbe1e
commit cac73ba
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 9 deletions.
diff --git a/cli/src/commands/verify_constraints.rs b/cli/src/commands/verify_constraints.rs
@@ -61,7 +61,7 @@ pub struct VerifyConstraintsCmd {
     pub field: Field,
 
     /// Verbosity (-v, -vv)
-    #[arg(short, long, action = clap::ArgAction::Count, help = "Increase verbosity level")] 
+    #[arg(short, long, action = clap::ArgAction::Count, help = "Increase verbosity level")]
     pub verbose: u8, // Using u8 to hold the number of `-v`
 }
 
@@ -73,12 +73,12 @@ impl VerifyConstraintsCmd {
         type GL = Goldilocks;
 
         let debug_mode = match self.verbose {
-             0 => 1, // Default to Error
-             1 => 2, // -v
-             2 => 3, // -vv _ => log::LevelFilter::Trace, 
-             _ => 1
+            0 => 1, // Default to Error
+            1 => 2, // -v
+            2 => 3, // -vv _ => log::LevelFilter::Trace,
+            _ => 1,
         };
-        
+
         let _valid_constraints = match self.field {
             Field::Goldilocks => ProofMan::<GL>::generate_proof(
                 self.witness_lib.clone(),

diff --git a/pilout/src/pilout_proxy.rs b/pilout/src/pilout_proxy.rs
@@ -62,7 +62,11 @@ impl PilOutProxy {
     }
 
     pub fn num_stages(&self) -> u32 {
-        self.pilout.num_challenges.len() as u32
+        if self.pilout.num_challenges.is_empty() {
+            1
+        } else {
+            self.pilout.num_challenges.len() as u32
+        }
     }
 
     pub fn num_rows(&self, subproof_id: usize, air_id: usize) -> usize {

diff --git a/proofman/src/proofman.rs b/proofman/src/proofman.rs
@@ -135,7 +135,7 @@ impl<F: Field + 'static> ProofMan<F> {
             log::info!("{}: <-- Verifying constraints", Self::MY_NAME);
 
             witness_lib.debug(&pctx, &ectx, &sctx);
-            
+
             let constraints = Self::verify_constraints(&mut provers, &mut pctx);
 
             let mut valid_constraints = true;
@@ -227,7 +227,9 @@ impl<F: Field + 'static> ProofMan<F> {
                     );
                 }
                 log::debug!("{}: ···   ", Self::MY_NAME);
-                if !valid_constraints_prover { valid_constraints = false; }
+                if !valid_constraints_prover {
+                    valid_constraints = false;
+                }
             }
 
             log::info!("{}: <-- Checking global constraints", Self::MY_NAME);