diff options
| -rw-r--r-- | gn2/wqflask/correlation/corr_scatter_plot.py | 6 |
1 files changed, 4 insertions, 2 deletions
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']) |