Documentation

[zonca]

It is nice you have docstrings in the functions.
It would be useful to use Sphynx to automatically extract those docstrings and build documentation, see for example PySM:
https://github.com/galsci/pysm/tree/main/docs

You can configure readthedocs so that at documentation is built at every commit: https://docs.readthedocs.io/en/stable/tutorial/index.html