Cleanup `astor` output to match `codegen` output.

The default `astor` output messes up the function signature docs for many docs without a bit of cleanup.

With this change the only differences I see are parens around lambdas and math expressions in default arguments.