Format docstrings in train_extensions/window_flip.py