diff options
| -rw-r--r-- | wqflask/wqflask/correlation/rust_correlation.py | 8 |
1 files changed, 2 insertions, 6 deletions
diff --git a/wqflask/wqflask/correlation/rust_correlation.py b/wqflask/wqflask/correlation/rust_correlation.py index df030e72..88133b31 100644 --- a/wqflask/wqflask/correlation/rust_correlation.py +++ b/wqflask/wqflask/correlation/rust_correlation.py @@ -98,12 +98,8 @@ def chunk_dataset(dataset, steps, name): for i in range(0, len(dataset), steps): matrix = list(dataset[i:i + steps]) - trait_name = traits_name_dict[matrix[0][0]] - - strains = [trait_name] + [str(value) - for (trait_name, strain, value) in matrix] - results.append(",".join(strains)) - + results.append([traits_name_dict[matrix[0][0]]] + [str(value) + for (trait_name, strain, value) in matrix]) return results |