Add some missing type annotations in Python code

This list appears when running

```shell
pixi run docs_serve
```

To preview the documentation.

<img width="1082" height="230" alt="Image" src="https://github.com/user-attachments/assets/5f7a4c9f-2ade-476d-ae75-6f93de403dfd" />