about summary refs log tree commit diff
diff options
context:
space:
mode:
-rw-r--r--gn3/db/traits.py18
1 files changed, 9 insertions, 9 deletions
diff --git a/gn3/db/traits.py b/gn3/db/traits.py
index f66ead3..6c31a4d 100644
--- a/gn3/db/traits.py
+++ b/gn3/db/traits.py
@@ -338,8 +338,7 @@ def retrieve_trait_info(
         lambda ti: load_qtl_info(qtl, trait_type, ti, conn),
         lambda ti: set_homologene_id_field(trait_type, ti, conn),
         lambda ti: {"trait_type": trait_type, **ti},
-        lambda ti: {**trait, **ti},
-        set_haveinfo_field)
+        lambda ti: {**trait, **ti})
 
     trait_post_processing_functions_table = {
         "Publish": compose(
@@ -353,8 +352,7 @@ def retrieve_trait_info(
     }
 
     retrieve_info = compose(
-        trait_post_processing_functions_table[trait_type],
-        trait_info_function_table[trait_type])
+        set_haveinfo_field, trait_info_function_table[trait_type])
 
     trait_dataset = retrieve_trait_dataset(trait_type, trait, threshold, conn)
     trait_info = retrieve_info(
@@ -364,8 +362,10 @@ def retrieve_trait_info(
             "trait_dataset_name": trait_dataset["dataset_name"]
         },
         conn)
-    return {
-        **trait_info,
-        "db": {**trait["db"], **trait_dataset},
-        "riset": trait_dataset["riset"]
-    }
+    if trait_info["haveinfo"]:
+        return {
+            **trait_post_processing_functions_table[trait_type](trait_info),
+            "db": {**trait["db"], **trait_dataset},
+            "riset": trait_dataset["riset"]
+        }
+    return trait_info