fix

KarrLab · Jul 13, 2020 · f4decec · f4decec
1 parent fd93ebf
commit f4decec
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/datanator_query_python/util/mongo_util.py b/datanator_query_python/util/mongo_util.py
@@ -180,7 +180,8 @@ def build_taxon_object(self, _id, _format="tax_id"):
         obj = self.client["datanator-test"]["taxon_tree"].find_one({_format: _id},
                                                                     projection={"canon_anc_ids": 1,
                                                                                 "canon_anc_names": 1,
-                                                                                "tax_name": 1})
+                                                                                "tax_name": 1,
+                                                                                "tax_id": 1})
         if obj is None:
             return {}
         else:
@@ -190,7 +191,7 @@ def build_taxon_object(self, _id, _format="tax_id"):
             for _id, name in zip(canon_anc_ids, canon_anc_names):
                 canon_ancestors.append({"ncbi_taxonomy_id": _id,
                                         "name": name})
-            return {"ncbi_taxonomy_id": taxon_id,
+            return {"ncbi_taxonomy_id": obj["tax_id"],
                     "name": obj["tax_name"],
                     "canon_ancestors": canon_ancestors}