about summary refs log tree commit diff
diff options
context:
space:
mode:
-rw-r--r--wqflask/wqflask/correlation/rust_correlation.py2
1 files changed, 1 insertions, 1 deletions
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 {}