fix: correctly write GWAS sumstats

Accidentally re-wrote everything every time
tatonetti-lab · Jul 24, 2024 · 4f46fa1 · 4f46fa1
1 parent 60128ed
commit 4f46fa1
Showing 1 changed file with 23 additions and 20 deletions.
diff --git a/src/io/gwas.rs b/src/io/gwas.rs
@@ -179,16 +179,18 @@ fn write_rows<W: std::io::Write>(
     add_header: bool,
     write_phenotype_id: bool,
 ) -> Result<()> {
-    if add_header && write_phenotype_id {
-        writer.write_record([
-            "phenotype_id",
-            "variant_id",
-            "beta",
-            "std_error",
-            "t_stat",
-            "neg_log_p_value",
-            "sample_size",
-        ])?;
+    if write_phenotype_id {
+        if add_header {
+            writer.write_record([
+                "phenotype_id",
+                "variant_id",
+                "beta",
+                "std_error",
+                "t_stat",
+                "neg_log_p_value",
+                "sample_size",
+            ])?;
+        }
         for i in 0..results.variant_ids.len() {
             writer.write_record(&[
                 results.projection_ids[i].clone(),
@@ -200,15 +202,17 @@ fn write_rows<W: std::io::Write>(
                 results.sample_sizes[i].to_string(),
             ])?;
         }
-    } else if add_header {
-        writer.write_record([
-            "variant_id",
-            "beta",
-            "std_error",
-            "t_stat",
-            "neg_log_p_value",
-            "sample_size",
-        ])?;
+    } else {
+        if add_header {
+            writer.write_record([
+                "variant_id",
+                "beta",
+                "std_error",
+                "t_stat",
+                "neg_log_p_value",
+                "sample_size",
+            ])?;
+        }
         for i in 0..results.variant_ids.len() {
             writer.write_record(&[
                 results.variant_ids[i].clone(),
@@ -220,7 +224,6 @@ fn write_rows<W: std::io::Write>(
             ])?;
         }
     }
-
     Ok(())
 }