Hi @mwilliamson! Would you be open to PRs that add type annotations? Thanks for writing this library!