formatting

Author: MahmoudAshraf97

ctc_forced_aligner/align.py

@@ -61,8 +61,8 @@ def cli():
         default="edges",
         choices=["segment", "edges"],
         help="The frequency of the <star> token in the text."
-        "Star token increases the accuracy of the alignment but also increases segment fragmentation."
-        "segment adds <star> token after each segment."
+        "Star token increases the accuracy of the alignment but also increases segment"
+        "fragmentation. segment adds <star> token after each segment."
         "edges adds <star> token at the start and end of the text."
         "use --merge_threshold to merge segments that are closer than the threshold.",
     )
@@ -118,7 +118,8 @@ def cli():
         type=str,
         default=None,
         choices=["eager", "sdpa", "flash_attention_2", None],
-        help="Attention implementation for the model. It will chose the fastest implementation by default.",
+        help="Attention implementation for the model. "
+        "It will chose the fastest implementation by default.",
     )
 
     parser.add_argument(

ctc_forced_aligner/text_utils.py

@@ -73,8 +73,12 @@ def text_normalize(
     normalized_text = re.sub(delete_patten, "", normalized_text)
 
     # Remove words containing only digits
-    # We check for 3 cases  a)text starts with a number b) a number is present somewhere in the middle of the text c) the text ends with a number
-    # For each case we use lookaround regex pattern to see if the digit pattern in preceded and followed by whitespaces, only then we replace the numbers with space
+    # We check for 3 cases:
+    #   a)text starts with a number
+    #   b) a number is present somewhere in the middle of the text
+    #   c) the text ends with a number
+    # For each case we use lookaround regex pattern to see if the digit pattern in preceded
+    # and followed by whitespaces, only then we replace the numbers with space
     # The lookaround enables overlapping pattern matches to be replaced
 
     if remove_numbers: