Typo in README file

[simoncorrea]

where it says TRRosettaDatset it's supposed to be TRRosettaDataset

from tape import TRRosetta
from tape.datasets import TRRosettaDataset

# Download data and place it under `<data_path>/trrosetta`

train_data = TRRosettaDatset('<data_path>', 'train')  # will subsample MSAs  #  <-- Here
valid_data = TRRosettaDatset('<data_path>', 'valid')  # will not subsample MSAs  <-- Here

model = TRRosetta.from_pretrained('xaa')  # valid choices are 'xaa', 'xab', 'xac', 'xad', 'xae'. Each corresponds to one of the ensemble models.

batch = train_data.collate_fn([train_data[0]])
loss, predictions = model(**batch)

[rmrao]

Oops - thanks for pointing it out!