Error when setting format of dataset to torch

[lhoestq]

Hi ! We fixed this issue on datasets on the main branch and we’ll do a new release soon

In the meantime you can also install datasetsfrom source if you want