aboutsummaryrefslogtreecommitdiff
path: root/gn3/db
diff options
context:
space:
mode:
Diffstat (limited to 'gn3/db')
-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