I see this in an NLP python program:

```
from datasets import load_dataset

train_set = load_dataset('wmt16', 'de-en', split='train')
train_size = train_set.num_rows
```

which python package I should install to use it? The response should be in all capital letters.
user