put ascii in if condition

jamesmkrieger · Nov 8, 2023 · f6baf7c · f6baf7c
1 parent 2adbe8e
commit f6baf7c
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/prody/proteins/ciffile.py b/prody/proteins/ciffile.py
@@ -434,13 +434,8 @@ def _parseMMCIFLines(atomgroup, lines, model, chain, subset,
                 continue
 
         alt = line.split()[fields['label_alt_id']]
-        if alt not in which_altlocs and which_altlocs != 'all':
-            try:
-                alt = ascii_uppercase[int(alt)-1]
-            except TypeError:
-                if not warnedAltloc:
-                    LOGGER.warn('alt {0} not in which_altlocs {1}'.format(alt, which_altlocs))
-                continue
+        if not (alt in which_altlocs or ascii_uppercase[int(alt)-1] in which_altlocs) and which_altlocs != 'all':
+            continue
 
         if alt == '.':
             alt = ' '