Importing RNA_SS_data into ufold_train.py

[charlottem-rna]

Hi,

I got this error:

$ python ufold_train.py --train_files TS1 TS2
#####Stage 1#####
Here is the configuration of this run: 
Munch({'gpu': '3', 'u_net_d': 10, 'BATCH_SIZE': 1, 'batch_size_stage_1': 1, 'OUT_STEP': 100, 'LOAD_MODEL': True, 'rho_per_position': 'matrix', 'data_type': 'rnastralign_all_600', 'model_type': 'pretrained', 'epoches_first': 100, 'step_gamma': 1, 'k': 1})
Loading dataset:  TS1
Loading dataset:  TS2
Data Loading Done!!!
Shape of data_list[0].pairs: (98, 268, 2)
Shape of data_list[1].pairs: (38, 268, 2)
start training...
Traceback (most recent call last):
  File "<string>", line 1, in <module>
  File "/home/exx/miniconda3/envs/UFold/lib/python3.11/multiprocessing/spawn.py", line 122, in spawn_main
    exitcode = _main(fd, parent_sentinel)
               ^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/exx/miniconda3/envs/UFold/lib/python3.11/multiprocessing/spawn.py", line 132, in _main
    self = reduction.pickle.load(from_parent)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
AttributeError: Can't get attribute 'RNA_SS_data' on <module '__mp_main__' from '/data/RUDD/RUDD/2d-prediction/charlotte/UFold/ufold_train.py'>
^CTraceback (most recent call last):
  File "/data/RUDD/RUDD/2d-prediction/charlotte/UFold/ufold_train.py", line 228, in <module>
    main()
  File "/data/RUDD/RUDD/2d-prediction/charlotte/UFold/ufold_train.py", line 218, in main
    train(contact_net,train_merge_generator,epoches_first)
  File "/data/RUDD/RUDD/2d-prediction/charlotte/UFold/ufold_train.py", line 61, in train
    for contacts, seq_embeddings, matrix_reps, seq_lens, seq_ori, seq_name in train_merge_generator:
  File "/home/exx/miniconda3/envs/UFold/lib/python3.11/site-packages/torch/utils/data/dataloader.py", line 441, in __iter__
    return self._get_iterator()
           ^^^^^^^^^^^^^^^^^^^^
  File "/home/exx/miniconda3/envs/UFold/lib/python3.11/site-packages/torch/utils/data/dataloader.py", line 388, in _get_iterator
    return _MultiProcessingDataLoaderIter(self)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/exx/miniconda3/envs/UFold/lib/python3.11/site-packages/torch/utils/data/dataloader.py", line 1042, in __init__
    w.start()
  File "/home/exx/miniconda3/envs/UFold/lib/python3.11/multiprocessing/process.py", line 121, in start
    self._popen = self._Popen(self)
                  ^^^^^^^^^^^^^^^^^
  File "/home/exx/miniconda3/envs/UFold/lib/python3.11/multiprocessing/context.py", line 224, in _Popen
    return _default_context.get_context().Process._Popen(process_obj)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/exx/miniconda3/envs/UFold/lib/python3.11/multiprocessing/context.py", line 288, in _Popen
    return Popen(process_obj)
           ^^^^^^^^^^^^^^^^^^
  File "/home/exx/miniconda3/envs/UFold/lib/python3.11/multiprocessing/popen_spawn_posix.py", line 32, in __init__
    super().__init__(process_obj)
  File "/home/exx/miniconda3/envs/UFold/lib/python3.11/multiprocessing/popen_fork.py", line 19, in __init__
    self._launch(process_obj)
  File "/home/exx/miniconda3/envs/UFold/lib/python3.11/multiprocessing/popen_spawn_posix.py", line 62, in _launch
    f.write(fp.getbuffer())
KeyboardInterrupt

...and then imported RNA_SS_data from ufold_test.py in ufold_train.py to effectively run training; is this the correct instance from which RNA_SS_data should be imported, or is there a different location it should be imported from, and it happened to be incorrectly named?