chore: format submodule/contract (#3286)

noir-lang · Oct 26, 2023 · 85a4ed3 · 85a4ed3
1 parent f31da68
commit 85a4ed3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/tooling/nargo_fmt/src/visitor/expr.rs b/tooling/nargo_fmt/src/visitor/expr.rs
@@ -192,7 +192,7 @@ impl FmtVisitor<'_> {
 
                 self.format_if(*if_expr)
             }
-            ExpressionKind::Variable(_) | ExpressionKind::Lambda(_) => self.slice(span).to_string(),
+            ExpressionKind::Lambda(_) | ExpressionKind::Variable(_) => self.slice(span).to_string(),
             ExpressionKind::Error => unreachable!(),
         }
     }

diff --git a/tooling/nargo_fmt/src/visitor/stmt.rs b/tooling/nargo_fmt/src/visitor/stmt.rs
@@ -55,7 +55,9 @@ impl super::FmtVisitor<'_> {
 
                     self.push_rewrite(constrain, span);
                 }
-                StatementKind::Assign(_) | StatementKind::For(_) => self.format_missing(span.end()),
+                StatementKind::For(_) | StatementKind::Assign(_) => {
+                    self.push_rewrite(self.slice(span).to_string(), span);
+                }
                 StatementKind::Error => unreachable!(),
             }