I watched this video by Anthony Sottile about a neat tool for auto-generating function annotations: auto-type-annotate. It is actually inspired by unmaintained tool called PyAnnotate.
Inspired by it, I added a custom VS Code shortcut: now when I press Cmd+K, then Cmd+M, annotations magically appear on [some] functions!