only get and set acs ind if hasattr

jamesmkrieger · Nov 6, 2023 · 00434d5 · 00434d5
1 parent 27e38ba
commit 00434d5
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/prody/proteins/pdbfile.py b/prody/proteins/pdbfile.py
@@ -1243,7 +1243,9 @@ def writePDBStream(stream, atoms, csets=None, **kwargs):
         NB: ChimeraX seems to prefer hybrid36 and may have problems with hexadecimal.
     :type hybrid36: bool
     """
-    initialACSI = atoms.getACSIndex()
+    if hasattr(atoms, 'getACSIndex'):
+        initialACSI = atoms.getACSIndex()
+
     renumber = kwargs.get('renumber', True)
 
     remark = str(atoms)
@@ -1614,7 +1616,8 @@ def writePDBStream(stream, atoms, csets=None, **kwargs):
 
     write('END   ' + " "*74 + '\n')
 
-    atoms.setACSIndex(initialACSI)
+    if hasattr(atoms, 'setACSIndex'):
+        atoms.setACSIndex(initialACSI)
 
 writePDBStream.__doc__ += _writePDBdoc