This usually happens if you use a script without using
if __name__ == "__main__":
    ...

This is because you don’t want your processes to be created when your module is imported.
Can you try again using this ?
from datasets import load_dataset

if __name__ == "__main__":
    dataset = load_dataset(...)
    ...