From 1d37d4514e2f60e16be23aa70c03948158e7b383 Mon Sep 17 00:00:00 2001
From: zsloan
Date: Tue, 11 Oct 2022 21:26:05 +0000
Subject: Only compute Lit/Tissue corr when main trait's dataset is of type
 ProbeSet

---
 wqflask/wqflask/correlation/rust_correlation.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

(limited to 'wqflask')

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),
-- 
cgit v1.2.3