[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

nemo_text_processing/inverse_text_normalization/en/taggers/date.py

@@ -147,12 +147,7 @@ class DateFst(GraphFst):
         input_case: accepting either "lower_cased" or "cased" input.
     """
 
-    def __init__(
-        self,
-        ordinal: GraphFst,
-        input_case: str,
-        project_input: bool = False
-    ):
+    def __init__(self, ordinal: GraphFst, input_case: str, project_input: bool = False):
         super().__init__(name="date", kind="classify", project_input=project_input)
 
         ordinal_graph = ordinal.graph

nemo_text_processing/inverse_text_normalization/en/taggers/decimal.py

@@ -83,12 +83,7 @@ class DecimalFst(GraphFst):
         input_case: accepting either "lower_cased" or "cased" input.
     """
 
-    def __init__(
-        self,
-        cardinal: GraphFst,
-        input_case: str = INPUT_LOWER_CASED,
-        project_input: bool = False
-    ):
+    def __init__(self, cardinal: GraphFst, input_case: str = INPUT_LOWER_CASED, project_input: bool = False):
         super().__init__(name="decimal", kind="classify", project_input=project_input)
 
         cardinal_graph = cardinal.graph_no_exception

nemo_text_processing/inverse_text_normalization/en/taggers/measure.py

@@ -43,11 +43,7 @@ class MeasureFst(GraphFst):
     """
 
     def __init__(
-        self,
-        cardinal: GraphFst,
-        decimal: GraphFst,
-        input_case: str = INPUT_LOWER_CASED,
-        project_input: bool = False
+        self, cardinal: GraphFst, decimal: GraphFst, input_case: str = INPUT_LOWER_CASED, project_input: bool = False
     ):
         super().__init__(name="measure", kind="classify", project_input=project_input)
 

nemo_text_processing/inverse_text_normalization/en/taggers/money.py

@@ -45,11 +45,7 @@ class MoneyFst(GraphFst):
     """
 
     def __init__(
-        self,
-        cardinal: GraphFst,
-        decimal: GraphFst,
-        input_case: str = INPUT_LOWER_CASED,
-        project_input: bool = False
+        self, cardinal: GraphFst, decimal: GraphFst, input_case: str = INPUT_LOWER_CASED, project_input: bool = False
     ):
         super().__init__(name="money", kind="classify", project_input=project_input)
         # quantity, integer_part, fractional_part, currency

nemo_text_processing/inverse_text_normalization/en/taggers/ordinal.py

@@ -36,12 +36,7 @@ class OrdinalFst(GraphFst):
         input_case: accepting either "lower_cased" or "cased" input.
     """
 
-    def __init__(
-        self,
-        cardinal: GraphFst,
-        input_case: str = INPUT_LOWER_CASED,
-        project_input: bool = False
-    ):
+    def __init__(self, cardinal: GraphFst, input_case: str = INPUT_LOWER_CASED, project_input: bool = False):
         super().__init__(name="ordinal", kind="classify", project_input=project_input)
 
         cardinal_graph = cardinal.graph_no_exception

nemo_text_processing/inverse_text_normalization/en/taggers/telephone.py

@@ -75,12 +75,7 @@ class TelephoneFst(GraphFst):
         input_case: accepting either "lower_cased" or "cased" input.
     """
 
-    def __init__(
-        self,
-        cardinal: GraphFst,
-        input_case: str = INPUT_LOWER_CASED,
-        project_input: bool = False
-    ):
+    def __init__(self, cardinal: GraphFst, input_case: str = INPUT_LOWER_CASED, project_input: bool = False):
         super().__init__(name="telephone", kind="classify", project_input=project_input)
         # country code, number_part, extension
         digit_to_str = (

nemo_text_processing/inverse_text_normalization/en/taggers/tokenize_and_classify.py

@@ -35,8 +35,8 @@
     GraphFst,
     delete_extra_space,
     delete_space,
-    generator_main,
     generate_far_filename,
+    generator_main,
 )
 from nemo_text_processing.utils.logging import logger
 
@@ -75,7 +75,7 @@ def __init__(
                 operation="tokenize_and_classify",
                 project_input=project_input,
                 input_case=input_case,
-                whitelist_file=whitelist
+                whitelist_file=whitelist,
             )
         if not overwrite_cache and far_file and os.path.exists(far_file):
             self.fst = pynini.Far(far_file, mode="r")["tokenize_and_classify"]
@@ -91,12 +91,18 @@ def __init__(
             decimal = DecimalFst(cardinal, input_case=input_case, project_input=project_input)
             decimal_graph = decimal.fst
 
-            measure_graph = MeasureFst(cardinal=cardinal, decimal=decimal, input_case=input_case, project_input=project_input).fst
+            measure_graph = MeasureFst(
+                cardinal=cardinal, decimal=decimal, input_case=input_case, project_input=project_input
+            ).fst
             date_graph = DateFst(ordinal=ordinal, input_case=input_case, project_input=project_input).fst
             word_graph = WordFst(project_input=project_input).fst
             time_graph = TimeFst(input_case=input_case, project_input=project_input).fst
-            money_graph = MoneyFst(cardinal=cardinal, decimal=decimal, input_case=input_case, project_input=project_input).fst
-            whitelist_graph = WhiteListFst(input_file=whitelist, input_case=input_case, project_input=project_input).fst
+            money_graph = MoneyFst(
+                cardinal=cardinal, decimal=decimal, input_case=input_case, project_input=project_input
+            ).fst
+            whitelist_graph = WhiteListFst(
+                input_file=whitelist, input_case=input_case, project_input=project_input
+            ).fst
             punct_graph = PunctuationFst(project_input=project_input).fst
             electronic_graph = ElectronicFst(input_case=input_case, project_input=project_input).fst
             telephone_graph = TelephoneFst(cardinal, input_case=input_case, project_input=project_input).fst

nemo_text_processing/inverse_text_normalization/en/taggers/whitelist.py

@@ -40,12 +40,7 @@ class WhiteListFst(GraphFst):
         input_case: accepting either "lower_cased" or "cased" input.
     """
 
-    def __init__(
-        self,
-        input_case: str = INPUT_LOWER_CASED,
-        input_file: str = None,
-        project_input: bool = False
-    ):
+    def __init__(self, input_case: str = INPUT_LOWER_CASED, input_file: str = None, project_input: bool = False):
         super().__init__(name="whitelist", kind="classify", project_input=project_input)
 
         if input_file is None:

nemo_text_processing/inverse_text_normalization/en/verbalizers/measure.py

@@ -29,12 +29,7 @@ class MeasureFst(GraphFst):
         cardinal: CardinalFst
     """
 
-    def __init__(
-        self,
-        decimal: GraphFst,
-        cardinal: GraphFst,
-        project_input: bool = False
-    ):
+    def __init__(self, decimal: GraphFst, cardinal: GraphFst, project_input: bool = False):
         super().__init__(name="measure", kind="verbalize", project_input=project_input)
         optional_sign = pynini.closure(pynini.cross("negative: \"true\"", "-"), 0, 1)
         unit = (

nemo_text_processing/inverse_text_normalization/en/verbalizers/verbalize_final.py

@@ -24,8 +24,8 @@
     GraphFst,
     delete_extra_space,
     delete_space,
-    generator_main,
     generate_far_filename,
+    generator_main,
 )
 from nemo_text_processing.utils.logging import logger
 
@@ -41,25 +41,16 @@ class VerbalizeFinalFst(GraphFst):
         overwrite_cache: set to True to overwrite .far files
     """
 
-    def __init__(
-        self,
-        project_input: bool = False,
-        cache_dir: str = None,
-        overwrite_cache: bool = False
-    ):
+    def __init__(self, project_input: bool = False, cache_dir: str = None, overwrite_cache: bool = False):
         super().__init__(name="verbalize_final", kind="verbalize")
 
         far_file = None
         if cache_dir is not None and cache_dir != "None":
             os.makedirs(cache_dir, exist_ok=True)
             far_file = generate_far_filename(
-                language="en",
-                mode="itn",
-                cache_dir=cache_dir,
-                operation="verbalize",
-                project_input=project_input
+                language="en", mode="itn", cache_dir=cache_dir, operation="verbalize", project_input=project_input
             )
-        
+
         if not overwrite_cache and far_file and os.path.exists(far_file):
             self.fst = pynini.Far(far_file, mode="r")["verbalize"]
             logger.info(f'VerbalizeFinalFst graph was restored from {far_file}.')
@@ -77,7 +68,7 @@ def __init__(
                 + pynutil.delete("}")
             )
             graph = delete_space + pynini.closure(graph + delete_extra_space) + graph + delete_space
-            
+
             self.fst = graph.optimize()
             if far_file:
                 generator_main(far_file, {"verbalize": self.fst})