🐛 map not working

[astariul]

I'm trying to run a basic example (mapping function to add a prefix).
Here is the colab notebook I'm using.

import nlp

dataset = nlp.load_dataset('squad', split='validation[:10%]')

def test(sample):
    sample['title'] = "test prefix @@@ " + sample["title"]
    return sample

print(dataset[0]['title'])
dataset.map(test)
print(dataset[0]['title'])

Output :

Super_Bowl_50
Super_Bowl_50

Expected output :

Super_Bowl_50
test prefix @@@ Super_Bowl_50

[astariul]

I didn't assign the output 🤦‍♂️

dataset.map(test)

should be :

dataset = dataset.map(test)