Specify dataset dtype
There was an issue in `datasets` <1.3.0 where the datasets type wouldn't be kept. Since this bug was patched, we have to specify the correct type for the dataset.

Co-authored-by: Quentin Lhoest <lhoest.q@gmail.com>
