pub_soft/foundry
3
0
Fork 0
mirror of https://github.com/RosettaCommons/foundry.git synced 2026-06-04 13:24:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix atom14 scheme cases

Browse Source
This commit is contained in:
Raktim Mitra
2026-02-18 22:51:57 -08:00
committed by Raktim Mitra
parent 3578002956
commit 3d016cd389
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
models/rfd3/src/rfd3/trainer/trainer_utils.py
View File

@@ -264,10 +264,14 @@ def _readout_seq_from_struc(
ATOM_NAMES = ATOM23_ATOM_NAMES_DNA
if not cur_res_atom_array.is_dna[0]:
continue
if restype in STANDARD_RNA:
elif restype in STANDARD_RNA:
ATOM_NAMES = ATOM23_ATOM_NAMES_RNA
if not cur_res_atom_array.is_rna[0]:
continue
else:
#ATOM_NAMES = ATOM23_ATOM_NAMES_RNA
if not cur_res_atom_array.is_protein[0]:
continue
atom_name_idx_in_atom14_scheme = np.array(
[