diff --git a/src/boltz/data/filter/static/polymer.py b/src/boltz/data/filter/static/polymer.py index e7e128279..5d83e763d 100644 --- a/src/boltz/data/filter/static/polymer.py +++ b/src/boltz/data/filter/static/polymer.py @@ -141,6 +141,7 @@ def filter(self, structure: Structure) -> np.ndarray: res_start = chain["res_idx"] res_end = res_start + chain["res_num"] residues = structure.residues[res_start:res_end] + residues = residues[np.where(residues["is_standard"])] # Get c-alphas ca_ids = residues["atom_center"]