From 091b7934178f3543c0037664d5aefc0a66fa96fb Mon Sep 17 00:00:00 2001 From: zsloan Date: Thu, 21 Mar 2024 17:50:58 +0000 Subject: Fix bug where scatterplot didn't work for temp traits --- gn2/wqflask/correlation/corr_scatter_plot.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) (limited to 'gn2') diff --git a/gn2/wqflask/correlation/corr_scatter_plot.py b/gn2/wqflask/correlation/corr_scatter_plot.py index 59e9ac4a..ab130cc5 100644 --- a/gn2/wqflask/correlation/corr_scatter_plot.py +++ b/gn2/wqflask/correlation/corr_scatter_plot.py @@ -20,13 +20,15 @@ class CorrScatterPlot: def __init__(self, params): if "Temp" in params['dataset_1']: + temp_group = params['trait_1'].split("_")[2] self.dataset_1 = data_set.create_dataset( - dataset_name="Temp", dataset_type="Temp", group_name=params['dataset_1'].split("_")[1]) + dataset_name="Temp", dataset_type="Temp", group_name=temp_group) else: self.dataset_1 = data_set.create_dataset(params['dataset_1']) if "Temp" in params['dataset_2']: + temp_group = params['trait_2'].split("_")[2] self.dataset_2 = data_set.create_dataset( - dataset_name="Temp", dataset_type="Temp", group_name=params['dataset_2'].split("_")[1]) + dataset_name="Temp", dataset_type="Temp", group_name=temp_group) else: self.dataset_2 = data_set.create_dataset(params['dataset_2']) -- cgit v1.2.3