mypy errors

Since traitlets v5.2.0, mypy errors out with:
```py
from traitlets import Integer

# error: Module "traitlets" has no attribute "Integer"  [attr-defined]
#     from traitlets import Integer
#     ^
# Found 1 error in 1 file (checked 1 source file)
```