Hi ! What version of datasets are you using ?
Also you may want to check if your dataset is indeed in RAM or loaded from the disk by checking dataset.cache_files. If it’s empty, then your dataset is fully in RAM. If not, then it shows the path to the cached arrow files on disk that are loaded.