From fc8c196e148bbeb5908506e0206f0bc0cb2fbdda Mon Sep 17 00:00:00 2001 From: rvcas Date: Sun, 2 Mar 2025 17:12:40 -0500 Subject: [PATCH] chore: fix issues for Pi --- crates/aiken-lang/src/format.rs | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/crates/aiken-lang/src/format.rs b/crates/aiken-lang/src/format.rs index 6dcd3feb..ddee569d 100644 --- a/crates/aiken-lang/src/format.rs +++ b/crates/aiken-lang/src/format.rs @@ -463,7 +463,17 @@ impl<'comments> Formatter<'comments> { .. } => "fn" .to_doc() - .append(wrap_args(args.iter().map(|t| (self.annotation(t), false)))) + .append(wrap_args(args.iter().map(|t| { + let comments = self.pop_comments(t.location().start); + + let doc_comments = self.doc_comments(t.location().start); + + let doc = doc_comments.append(self.annotation(t)).group(); + + let doc = commented(doc, comments); + + (doc, false) + }))) .group() .append(" ->") .append(break_("", " ").append(self.annotation(retrn)).nest(INDENT)),