cleaning up

VirtualFlyBrain · Aug 12, 2024 · af30a83 · af30a83
1 parent d6cd571
commit af30a83
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/vfb_connect/neo/query_wrapper.py b/src/vfb_connect/neo/query_wrapper.py
@@ -410,7 +410,7 @@ def get_terms_by_xref(self, acc, db='', id_type=''):
             if key not in vfb_ids.keys():
                 warnings.warn("No VFB ID found for %s" % key)
             else:
-                ids_to_query.append(vfb_ids[key])
+                ids_to_query.extend(vfb_ids[key])
 
         # Retrieve term information for all IDs
         return self.get_TermInfo(ids_to_query)
@@ -482,7 +482,8 @@ def _get_Cached_TermInfo(self, short_forms: iter, summary=True):
         results=[]
         for short_form in short_forms:
             result = vfb_solr.search('id:' + short_form)
-            results.extend(self._serialize_solr_output(result))
+            if len(result.docs) > 0:
+                results.extend(self._serialize_solr_output(result))
         return results