Add type hints throughout the codebase and enforce

[vmenger]

Great to enforce type hints for readable code. Ruff can do this through flake8-annotations: https://docs.astral.sh/ruff/rules/#flake8-annotations-ann