From ee4436538b124b1cd311b396998b1d1d9eb641ef Mon Sep 17 00:00:00 2001
From: Frederick Muriuki Muriithi
Date: Tue, 16 Aug 2022 16:04:19 +0300
Subject: Merge dicts into single result

---
 wqflask/wqflask/correlation/rust_correlation.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

(limited to 'wqflask')

diff --git a/wqflask/wqflask/correlation/rust_correlation.py b/wqflask/wqflask/correlation/rust_correlation.py
index b93c306d..79d08a59 100644
--- a/wqflask/wqflask/correlation/rust_correlation.py
+++ b/wqflask/wqflask/correlation/rust_correlation.py
@@ -122,9 +122,12 @@ def __compute_lit_corr__(
         this_trait, this_dataset)
 
     with database_connector() as conn:
-        return dict(compute_all_lit_correlation(
-            conn=conn, trait_lists=list(geneid_dict.items()),
-            species=species, gene_id=this_trait_geneid))
+        return reduce(
+            lambda acc, lit: {**acc, **lit},
+            compute_all_lit_correlation(
+                conn=conn, trait_lists=list(geneid_dict.items()),
+                species=species, gene_id=this_trait_geneid),
+            {})
     return {}
 
 def compute_correlation_rust(
-- 
cgit v1.2.3