Merge pull request #241 from bamattsson/fixing_bug_in_atom_preprocessing

Fixing bug leading to atom graph containing a lot of incorrect atoms
This commit is contained in:
Jacob Silterra
2024-09-03 15:57:25 -04:00
committed by GitHub

View File

@@ -61,7 +61,7 @@ def batch_compute_dihedral_angles(sidechains):
def get_coords(prody_pdb):
resindices = sorted(set(prody_pdb.ca.getResindices()))
coords = np.zeros((len(resindices), 14, 3))
coords = np.full((len(resindices), 14, 3), np.nan)
for i, resind in enumerate(resindices):
sel = prody_pdb.select(f'resindex {resind}')
resname = sel.getResnames()[0]