Format docstrings in train_extensions/__init__.py
Format docstrings in termination_criteria/__init__.py