From 400320ea0063954f027fa918f77f7518b904c6ba Mon Sep 17 00:00:00 2001
From: zsloan
Date: Fri, 21 Oct 2022 23:31:56 +0000
Subject: Fix logic for determining samplelist in correlations

---
 wqflask/wqflask/correlation/correlation_gn3_api.py | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/wqflask/wqflask/correlation/correlation_gn3_api.py b/wqflask/wqflask/correlation/correlation_gn3_api.py
index 74796383..1af54867 100644
--- a/wqflask/wqflask/correlation/correlation_gn3_api.py
+++ b/wqflask/wqflask/correlation/correlation_gn3_api.py
@@ -159,17 +159,15 @@ def fetch_sample_data(start_vars, this_trait, this_dataset, target_dataset):
     corr_samples_group = start_vars["corr_samples_group"]
     if corr_samples_group == "samples_primary":
         sample_data = process_samples(
-            start_vars, this_dataset.group.all_samples_ordered())
+            start_vars, this_dataset.group.samplelist)
 
     elif corr_samples_group == "samples_other":
         sample_data = process_samples(
             start_vars, excluded_samples=this_dataset.group.samplelist)
 
     else:
-        sample_data = process_samples(start_vars)
-
-    sample_data = process_samples(
-        start_vars, this_dataset.group.all_samples_ordered())
+        sample_data = process_samples(start_vars,
+            this_dataset.group.all_samples_ordered())
 
     target_dataset.get_trait_data(list(sample_data.keys()))
     this_trait = retrieve_sample_data(this_trait, this_dataset)
-- 
cgit v1.2.3