Use Python type hints instead of docstring to document types

[snazzyfox]

Most data types are documented in docstrings. We should move them all to type hints because this will allow use to run static type checks, and as a benefit, catch undocumented variables.

Together with #60 this will allow us to enable more stringent static type checking.