Adding support for balanced line breaks?

[danijar]

Adding support for balanced line breaks?

[bwendling]

For the first one, see if SPLIT_COMPLEX_COMPREHENSION might help. (I think it will still split after the k: v part though.) As for the second example, you might try adjusting SPLIT_PENALTY_AFTER_OPENING_BRACKET. However, it may have other side effects.

[danijar]

Thanks for the suggestions!

Example 1

With split_complex_comprehension = true, it generates:

donated = {
    k: v
    for k, v in very_long_dict_name.items()
    if k not in self.copy_keys}

I think this looks better. Not ideal, but I assume having a line balancing heuristic (i.e. avoiding very short lines) might be too tricky to consider?

Example 2

I tried different values of split_penalty_after_opening_bracket (default is 300, I tried 1, 100, 1000, 10000). None of those changed the generated output. I think it's probably because both breakpoints (the first one YAPF chooses and the second one that would be desired) are after an opening brackets, so the penalty would be the same either way?

Perhaps the current penalty system would already make it feasible to achieve the desired balancing behavior by adding a small penalty for based on line shortness (i.e. 79 minus line length, summed over all lines in the formatted expression)?