Format docstrings in training_algorithms/bgd.py