Hi !
First, the output 2 that you got ValueError: Keys mismatch comes from a cache issue that has been fixed on the master branch of datasets recently, see Backwards compatibility broken for cached datasets that use `.filter()` · Issue #2943 · huggingface/datasets · GitHub
Then, regarding the other output, can you check the logs if it reloads previously computed results from the cache or not ?