Clean up formatting in train_extensions/__init__.py