From 278621273f01435472a49fb84f77d9dbf581abae Mon Sep 17 00:00:00 2001
From: Frederick Muriuki Muriithi
Date: Fri, 28 Oct 2022 16:15:17 +0300
Subject: Add `sample_data` argument to function call.

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

diff --git a/wqflask/wqflask/correlation/rust_correlation.py b/wqflask/wqflask/correlation/rust_correlation.py
index 4b7eedc0..a2f842bc 100644
--- a/wqflask/wqflask/correlation/rust_correlation.py
+++ b/wqflask/wqflask/correlation/rust_correlation.py
@@ -116,7 +116,9 @@ def compute_top_n_sample(start_vars, dataset, trait_list):
     def __fetch_sample_ids__(samples_vals, samples_group):
         all_samples = json.loads(samples_vals)
         sample_data = get_sample_corr_data(
-            sample_type=samples_group, all_samples=all_samples,
+            sample_type=samples_group,
+            sample_data=json.loads(start_vars["sample_vals"]),
+            all_samples=all_samples,
             dataset_samples=dataset.group.all_samples_ordered())
 
         with database_connector() as conn:
-- 
cgit v1.2.3