Not necessary for this review, but in the future, it could be useful to standardize the docstrings. One example of a standard is numpydoc.
https://github.com/openjournals/joss-reviews/issues/6930