include f1 parents for correlation

1 files changed, 7 insertions, 0 deletions

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 {}