From d0c210dfb79ddf851d8cd721801c88df40f08d5a Mon Sep 17 00:00:00 2001
From: zsloan
Date: Wed, 24 Aug 2022 21:20:06 +0000
Subject: Limit literature correlation results to top N as specified in the
 form

---
 wqflask/wqflask/correlation/rust_correlation.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'wqflask')

diff --git a/wqflask/wqflask/correlation/rust_correlation.py b/wqflask/wqflask/correlation/rust_correlation.py
index 5c22efbf..2a90317b 100644
--- a/wqflask/wqflask/correlation/rust_correlation.py
+++ b/wqflask/wqflask/correlation/rust_correlation.py
@@ -212,7 +212,7 @@ def __compute_lit_corr__(
             lambda acc, lit: {**acc, **lit},
             compute_all_lit_correlation(
                 conn=conn, trait_lists=list(geneid_dict.items()),
-                species=species, gene_id=this_trait_geneid),
+                species=species, gene_id=this_trait_geneid)[:n_top],
             {})
     return {}
 
-- 
cgit v1.2.3