Format docstrings in utils/one_hot.py