TrRosetta example doesn't work

[GrayWasTaken]

The example code provided in the README.md for TrRosetta doesn't work and throws the following exception:

Traceback (most recent call last):
  File "...", line 1, in <module>
    msas_processed = tr_preprocessing.process(msas)
  File ".venv/lib/python3.10/site-packages/sequence_models/trRosetta_utils.py", line 254, in process
    x = F.one_hot(x, len(trR_ALPHABET)).unsqueeze(0).float()
TypeError: one_hot(): argument 'input' (position 1) must be Tensor, not list

code:

from sequence_models.trRosetta_utils import trRosettaPreprocessing, parse_a3m
from sequence_models.trRosetta import trRosetta
from sequence_models.constants import trR_ALPHABET

msas = parse_a3m(path_to_msa) # load in msas in a3m format
alphabet = trR_ALPHABET # load your alphabet order
tr_preprocessing = trRosettaPreprocessing(alphabet) # setup preprocessor for msa
msas_processed = tr_preprocessing.process(msas)