add thread option to prediction

Author: XiaotingChen

maxatac/analyses/predict.py

@@ -102,7 +102,7 @@ def run_prediction(args):
                  f"Output filename: {outfile_name_bigwig}"
                  )
 
-    with Pool(int(multiprocessing.cpu_count())) as p:
+    with Pool(args.threads) as p:
         forward_strand_predictions = p.starmap(make_stranded_predictions,
                                                [(regions_pool,
                                                  args.signal,

maxatac/utilities/parser.py

@@ -330,6 +330,12 @@ def get_parser():
                                 help="Skip calling peaks on prediction tracks"
                                 )
 
+    predict_parser.add_argument("--threads",
+                              dest="threads",
+                              type=int,
+                              default=24,
+                              help="Number of processes to run prediction in parallel. Default: 24"
+                              )
     #############################################
     # Train parser
     #############################################