remove data preprocessing for top n samples

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