mmtf fix

jamesmkrieger · Nov 8, 2023 · 2adbe8e · 2adbe8e
1 parent 0a00bfa
commit 2adbe8e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/prody/proteins/mmtffile.py b/prody/proteins/mmtffile.py
@@ -323,6 +323,9 @@ def set_info(atomgroup, mmtf_data,get_bonds=False,altloc_sel='A'):
     if altloc_sel != 'all':
         #mask out any unwanted alternative locations
         mask = (altlocs == '') | (altlocs == altloc_sel)
+
+    if np.all(mask == False):
+        mask = (altlocs == '') | (altlocs == altlocs[0])
 
     atomgroup.setCoords(coords[:,mask])
     atomgroup.setNames(atom_names[mask])