[ASDL cleanup] Add more type annotations