I am not sure, but I think you need to add a `py.typed` file to the package for mypy to use the typing from the PyPI package  https://mypy.readthedocs.io/en/stable/running_mypy.html#missing-library-stubs-or-py-typed-marker  