get_best_motif.py: added parameter cluster id

loosolab · Jan 9, 2019 · d979e16 · d979e16
1 parent 98ab76a
commit d979e16
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/bin/2.2_motif_estimation/get_best_motif.py b/bin/2.2_motif_estimation/get_best_motif.py
@@ -7,6 +7,7 @@ def parse_arguments():
     parser.add_argument("meme", help="Path to 'meme' file generated by GLAM2")
     parser.add_argument("output", help="Output file")
     parser.add_argument("num", help="Number of motifs parsed from file")
+    parser.add_argument("id", help="Cluster ID")
     args = parser.parse_args()
     return args
 
@@ -36,7 +37,7 @@ def main():
     break_header = "MOTIF  " + str(number)
 
     # Get cluster_id
-    cluster_id = os.path.splitext(args.meme)[0].split('/')[-2].split('_')[-1]
+    cluster_id = args.id
 
     # Pattern for motif header
     pattern = re.compile("^MOTIF\s{2}(\d)+")