ModuleNotFoundError: No module named 'datasets'

[zhongjian-zhang]

Hi, thanks for you repo. I meet a problem when run the train_CLM.py, the error information is that "ModuleNotFoundError: No module named 'datasets'". In the "LMs" package, I don't find the 'datasets' package as well as the "load_dataset" function.

Traceback (most recent call last):
File "/home/pycharm_project/TAG-Benchmark/LMs/train_CLM.py", line 32, in
import datasets
ModuleNotFoundError: No module named 'datasets'

[sktsherlock]

The datasets package is a library provided by Hugging Face for accessing and handling various NLP datasets. Before running train_CLM.py, you need to ensure that the datasets library is installed.
You can install the datasets library using the following command: pip install datasets
If you already have the datasets library installed but are still facing the issue, it might be due to an incompatible version. Try updating the datasets library to the latest version: pip install --upgrade datasets