about summary refs log tree commit diff
diff options
context:
space:
mode:
-rw-r--r--wqflask/wqflask/correlation/rust_correlation.py4
1 files changed, 1 insertions, 3 deletions
diff --git a/wqflask/wqflask/correlation/rust_correlation.py b/wqflask/wqflask/correlation/rust_correlation.py
index aee08cba..3330cce9 100644
--- a/wqflask/wqflask/correlation/rust_correlation.py
+++ b/wqflask/wqflask/correlation/rust_correlation.py
@@ -348,7 +348,7 @@ def compute_correlation_rust(
 
     if should_compute_all:
 
-        if corr_type == "sample":
+        if corr_type == "sample" and this_dataset.type == "ProbeSet":
 
             top_a = compute_top_n_tissue(
                 target_dataset, this_trait, results, method)
@@ -369,8 +369,6 @@ def compute_correlation_rust(
             top_a = compute_top_n_sample(
                 start_vars, target_dataset, list(results.keys()))
 
-            top_b = compute_top_n_lit(results, target_dataset, this_trait)
-
     return {
         "correlation_results": merge_results(
             results, top_a, top_b),