AttributeError: module 'mp_nerf.proteins' has no attribute 'ca_bb_fold'

[lipan6461188]

I want to train the rgn2-replica model with the following command:

export WANDB_MODE=offline
python scripts/train_rgn2.py \
    --device cuda:1 \
    --wb_entity xxxxx \
    --wb_proj rgn2 \
    --run_name RGN2X_vanillaLSTM_full_run \
    --min_len_valid 0 
    --xray 1

But an AttributeError occurred:

Traceback (most recent call last):
  File "scripts/train_rgn2.py", line 345, in <module>
    init_and_train(args)
  File "scripts/train_rgn2.py", line 134, in init_and_train
    results = run_train_schedule(dataloaders, embedder, config, args)
  File "scripts/train_rgn2.py", line 213, in run_train_schedule
    config=config,
  File "/home/lipan/usr/rgn2-replica/rgn2_replica/rgn2_trainers.py", line 432, in train
    config=config,
  File "/home/lipan/usr/rgn2-replica/rgn2_replica/rgn2_trainers.py", line 62, in batched_inference
    coords_rebuilt = mp_nerf.proteins.ca_bb_fold( ca_trace ) # beware extremes
AttributeError: module 'mp_nerf.proteins' has no attribute 'ca_bb_fold'

It seems that mp_nerf.proteins.ca_bb_fold is not defined in https://github.com/eleutherAI/mp_nerf