Fix output delimiter selection for mp script

- #103
aertslab · Jun 8, 2020 · 321d670 · 321d670
1 parent 264fc7e
commit 321d670
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/scripts/arboreto_with_multiprocessing.py b/scripts/arboreto_with_multiprocessing.py
@@ -13,7 +13,8 @@
 from arboreto.core import SGBM_KWARGS, RF_KWARGS, EARLY_STOP_WINDOW_LENGTH
 from arboreto.core import to_tf_matrix, target_gene_indices, infer_partial_network
 
-from pyscenic.cli.utils import load_exp_matrix
+from pyscenic.cli.utils import load_exp_matrix, suffixes_to_separator
+from pathlib import PurePath
 
 
 ################################################################################
@@ -126,5 +127,6 @@ def run_infer_partial_network(target_gene_index):
     end_time = time.time()
     print(f'Done in {end_time - start_time} seconds.', file=sys.stdout)
 
-    adj.to_csv(args.output, index=False, sep='\t')
+    extension = PurePath(args.output).suffixes
+    adj.to_csv(args.output, index=False, sep=suffixes_to_separator(extension))