Skip to content

Commit 989cfb8

Browse files
authored
Merge pull request #1623 from SciML/myb/pretty
Remove macro linnums in `readable_code`
2 parents 8ae89a2 + 633b526 commit 989cfb8

File tree

1 file changed

+17
-1
lines changed

1 file changed

+17
-1
lines changed

src/utils.jl

Lines changed: 17 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -102,8 +102,24 @@ function _readable_code(ex)
102102
end
103103
expr
104104
end
105+
106+
function rec_remove_macro_linenums!(expr)
107+
if expr isa Expr
108+
if expr.head === :macrocall
109+
expr.args[2] = nothing
110+
rec_remove_macro_linenums!(expr.args[3])
111+
else
112+
for ex in expr.args
113+
rec_remove_macro_linenums!(ex)
114+
end
115+
end
116+
end
117+
expr
118+
end
105119
function readable_code(expr)
106-
JuliaFormatter.format_text(string(Base.remove_linenums!(_readable_code(expr))))
120+
expr = Base.remove_linenums!(_readable_code(expr))
121+
rec_remove_macro_linenums!(expr)
122+
JuliaFormatter.format_text(string(expr), JuliaFormatter.SciMLStyle())
107123
end
108124

109125
function check_parameters(ps, iv)

0 commit comments

Comments
 (0)