From 8beccb5a46f30bb65e7eae294d4cdb17a051a806 Mon Sep 17 00:00:00 2001
From: AlexanderKabui
Date: Mon, 7 Nov 2022 18:10:26 +0300
Subject: include f1 parents for correlation

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

(limited to 'wqflask')

diff --git a/wqflask/wqflask/correlation/rust_correlation.py b/wqflask/wqflask/correlation/rust_correlation.py
index a2f842bc..3245d037 100644
--- a/wqflask/wqflask/correlation/rust_correlation.py
+++ b/wqflask/wqflask/correlation/rust_correlation.py
@@ -234,11 +234,18 @@ def __compute_sample_corr__(
     """Compute the sample correlations"""
     (this_dataset, this_trait, target_dataset, sample_data) = target_trait_info
 
+    if this_dataset.group.f1list !=None:
+        this_dataset.group.samplelist+= this_dataset.group.f1list
+
+    if this_dataset.group.parlist!= None:
+        this_dataset.group.samplelist+= this_dataset.group.parlist
+
     sample_data = get_sample_corr_data(
         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 {}
 
-- 
cgit v1.2.3