If psims is available, use its Unimod because it caches and vendors

mobiusklein · Oct 10, 2023 · 1bc5b35 · 1bc5b35
1 parent e2d4a8a
commit 1bc5b35
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pyteomics/proforma.py b/pyteomics/proforma.py
@@ -32,15 +32,18 @@
     np = None
 
 try:
-    from psims.controlled_vocabulary.controlled_vocabulary import (load_psimod, load_xlmod, load_gno, obo_cache)
+    from psims.controlled_vocabulary.controlled_vocabulary import (load_psimod, load_xlmod, load_gno, obo_cache, load_unimod)
+    _has_psims = True
 except ImportError:
     def _needs_psims(name):
         raise ImportError("Loading %s requires the `psims` library. To access it, please install `psims`" % name)
 
     load_psimod = partial(_needs_psims, 'PSIMOD')
     load_xlmod = partial(_needs_psims, 'XLMOD')
     load_gno = partial(_needs_psims, 'GNO')
+    load_unimod = partial(_needs_psims, 'UNIMOD')
     obo_cache = None
+    _has_psims = False
 
 _water_mass = calculate_mass("H2O")
 
@@ -357,6 +360,8 @@ def __init__(self, **kwargs):
         self.strict = kwargs.get("strict", True)
 
     def load_database(self):
+        if _has_psims:
+            return load_unimod()
         return Unimod()
 
     def resolve(self, name=None, id=None, **kwargs):