Yapf does not format some function

I want all my function to look like these
`def some_thing(
param_dict: dict,
depth_polar: np.ndarray,
intensity_polar: Optional[np.ndarray] = None,
) -> Tuple[np.ndarray, np.ndarray]:
pass`

And yapf format it well for most of the time but there is one function where it does not work :

`def another_thing(polar: np.ndarray,
circles: np.ndarray,
r_min_px: int,
dist_max_px: int = 10) -> Tuple[np.ndarray, np.ndarray, np.ndarray]:
pass`

Both functions are in the same file...

Here are my arguments usinge VS Code format on save :
--style
{based_on_style: pep8, column_limit: 120, DEDENT_CLOSING_BRACKETS: true}

I use VS Code and yapf v0.31.0

[yapf_example.zip](https://github.com/google/yapf/files/6273238/yapf_example.zip)