yapf: Tweak style

17 jobs for code_formatter in 10 minutes and 9 seconds (queued for 10 seconds)