proteincomponent: tweak conversions to/from openmm/rdkit

helps with some nonstandard residues previously giving KeyError AROMATIC will not affect serialised representations (i.e. keys) as these don't use this path
OpenFreeEnergy · Jul 18, 2023 · 5ab9ecc · 5ab9ecc
1 parent 239302a
commit 5ab9ecc
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/gufe/components/proteincomponent.py b/gufe/components/proteincomponent.py
@@ -27,6 +27,10 @@
     1: BondType.SINGLE,
     2: BondType.DOUBLE,
     3: BondType.TRIPLE,
+    app.Single: BondType.SINGLE,
+    app.Double: BondType.DOUBLE,
+    app.Triple: BondType.TRIPLE,
+    app.Aromatic: BondType.AROMATIC,
     None: BondType.UNSPECIFIED,
 }
 _BONDORDERS_RDKIT_TO_OPENMM = {
@@ -361,7 +365,8 @@ def chainkey(m):
             a1 = atom_lookup[bond.GetBeginAtomIdx()]
             a2 = atom_lookup[bond.GetEndAtomIdx()]
             top.addBond(a1, a2,
-                        order=_BONDORDERS_RDKIT_TO_OPENMM[bond.GetBondType()])
+                        order=_BONDORDERS_RDKIT_TO_OPENMM.get(
+                            bond.GetBondType(), None))
 
         return top