This library should have type annotations

[benbariteau]

What would you like to be added

Type annotations

Why is this needed

For using this library with a typechecker

[a-ungurianu]

I'm removing the last footprints (hopefully) of python2 in #678. This will allow us to write types in the py files directly. Lemme know if you wanna collaborate on this.