Lambdas are reformatted into (potentially) very long lines

[jjmaestro]

As I mentioned in #890, I have the following code:

def foo(foobar):
    return struct(
        bar = 123,
        some_very_long_line = lambda arg1, arg2, arg3:
            _some_other_method(foobar, arg1, arg2, arg3),
)

I was expecting buildifier to respect my line returns there but instead, it reformats the lambda in one very long line:

def foo(foobar):
    return struct(
        bar = 123,
        some_very_long_line = lambda arg1, arg2, arg3: _some_other_method(foobar, arg1, arg2, arg3),
)

The next-best-match is the following:

def foo(foobar):
    return struct(
        bar = 123,
        some_very_long_line = lambda arg1, arg2, arg3: _some_other_method(
            foobar, arg1, arg2, arg3
        ),
)

which gets reformatted to:

def foo(foobar):
    return struct(
        bar = 123,
        some_very_long_line = lambda arg1, arg2, arg3: _some_other_method(
            foobar,
            arg1,
            arg2,
            arg3,
        ),
)

This is OK-ish and it does make sense but it can also turn into a very long line "over 80+ chars".

Is there a way to get buildifier to respect a line return like I did and/or to break after the ':' in lambdas that would end up being a very long line?

Thanks!