STY: optimize: better whitespace alignment in a few lines in anneal.py