Reformatted source files with Black

[galenseilis]

Ran Black on the source code. None of the changes should affect run time (and hopefully improve readability and consistency).