Map function type hinting?

Is there an established method of adding type hinting to map/batched map functions?

This is mainly for other human readers to understand what the input/output row/batch should look like, but would be a “nice to have” if it also allowed IDE type checking.

I’m thinking a method to datasets.Features that generated a TypedDict object (with a row/batch version)?

Hi ! AFAIK people simply use regular python type hints right now (without links to the Features object)

Do you have an idea of how the map function could look like if there was something like this ? And how beneficial it would be ?