I think adding type annotations to attributes and functions of this package will make the development experience easier. https://docs.python.org/3/library/typing.html