Mark package as type-annotated.

Although the package uses type annotations extensively, it doesn't mark itself as such, causing issues to downstream users. I've added the necessary py.typed marker file.

While on it, I've also fixed a few typing errors and one of the PyLint warnings.