Cannot user load_dataset in Google colab

Sounds like an issue with huggingface_hub, can you try to udpate it ?

pip install -U huggingface_hub