aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--wqflask/wqflask/correlation/rust_correlation.py8
1 files changed, 5 insertions, 3 deletions
diff --git a/wqflask/wqflask/correlation/rust_correlation.py b/wqflask/wqflask/correlation/rust_correlation.py
index b2a0e3a0..4b7eedc0 100644
--- a/wqflask/wqflask/correlation/rust_correlation.py
+++ b/wqflask/wqflask/correlation/rust_correlation.py
@@ -231,10 +231,12 @@ def __compute_sample_corr__(
target_trait_info: tuple):
"""Compute the sample correlations"""
(this_dataset, this_trait, target_dataset, sample_data) = target_trait_info
- all_samples = json.loads(start_vars["sample_vals"])
+
sample_data = get_sample_corr_data(
- sample_type=start_vars["corr_samples_group"], all_samples=all_samples,
- dataset_samples=this_dataset.group.all_samples_ordered())
+ sample_type=start_vars["corr_samples_group"],
+ sample_data= json.loads(start_vars["sample_vals"]),
+ all_samples=this_dataset.group.all_samples_ordered(),
+ dataset_samples=this_dataset.group.samplelist)
if not bool(sample_data):
return {}