Load models before getting models

gyorilab · Jul 19, 2022 · 3be0c44 · 3be0c44
1 parent 54e14ba
commit 3be0c44
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/gilda/grounder.py b/gilda/grounder.py
@@ -311,6 +311,8 @@ def get_models(self):
             The list of entity texts for which a disambiguation model is
             available.
         """
+        if self.gilda_disambiguators is None:
+            self.gilda_disambiguators = load_gilda_models()
         return sorted(list(self.gilda_disambiguators.keys()))
 
     def get_names(self, db, id, status=None, source=None):