[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/text_normalization/ko/taggers/cardinal.py

@@ -50,27 +50,21 @@ def __init__(
         deterministic: bool = True,
         cache_dir: str = None,
         overwrite_cache: bool = False,
-        whitelist: str = None
+        whitelist: str = None,
     ):
         super().__init__(name="tokenize_and_classify", kind="classify", deterministic=deterministic)
 
         far_file = None
         if cache_dir is not None and cache_dir != "None":
             os.makedirs(cache_dir, exist_ok=True)
-            far_file = os.path.join(
-                cache_dir,
-                f"ko_tn_{deterministic}_tokenize.far"
-            )
+            far_file = os.path.join(cache_dir, f"ko_tn_{deterministic}_tokenize.far")
         if not overwrite_cache and far_file and os.path.exists(far_file):
             self.fst = pynini.Far(far_file, mode="r")["tokenize_and_classify"]
             logger.info(f"ClassifyFst.fst was restored from {far_file}.")
         else:
             cardinal = CardinalFst(deterministic=deterministic)
 
-            classify = pynini.union(
-
-                pynutil.add_weight(cardinal.fst, 1.1)
-            )
+            classify = pynini.union(pynutil.add_weight(cardinal.fst, 1.1))
 
             token = pynutil.insert("tokens { ") + classify + pynutil.insert(" }")
             tagger = pynini.closure(token, 1)

nemo_text_processing/text_normalization/ko/taggers/tokenize_and_classify.py

@@ -31,8 +31,8 @@ class VerbalizeFst(GraphFst):
 
     def __init__(self, deterministic: bool = True):
         super().__init__(name="verbalize", kind="verbalize", deterministic=deterministic)
-        
+
         cardinal = CardinalFst(deterministic=deterministic)
         cardinal_graph = cardinal.fst
 
-        self.fst = cardinal_graph.optimize()
+        self.fst = cardinal_graph.optimize()

nemo_text_processing/text_normalization/ko/verbalizers/verbalize.py

@@ -51,7 +51,7 @@ def __init__(self, deterministic: bool = True, cache_dir: str = None, overwrite_
             logger.info(f'VerbalizeFinalFst graph was restored from {far_file}.')
         else:
             verbalize = VerbalizeFst(deterministic=deterministic).fst
-            #word = WordFst(deterministic=deterministic).fst
+            # word = WordFst(deterministic=deterministic).fst
             types = verbalize
 
             if deterministic:

nemo_text_processing/text_normalization/ko/verbalizers/verbalize_final.py

@@ -22,12 +22,13 @@
 
 
 class TestCardinal:
-    normalizer_ko = Normalizer(lang='ko', cache_dir='export/ko_tn_grammars_lower_cased', overwrite_cache=False, input_case='lower_cased')
+    normalizer_ko = Normalizer(
+        lang='ko', cache_dir='export/ko_tn_grammars_lower_cased', overwrite_cache=False, input_case='lower_cased'
+    )
 
     @parameterized.expand(parse_test_case_file('ko/data_text_normalization/test_cases_cardinal.txt'))
     @pytest.mark.run_only_on('CPU')
     @pytest.mark.unit
     def test_norm(self, test_input, expected):
         preds = self.normalizer_ko.normalize(test_input)
         assert expected == preds
-

tests/nemo_text_processing/ko/test_cardinal.py

@@ -39,7 +39,7 @@ def parse_test_case_file(file_name: str):
     Prepares tests pairs for ITN and TN tests
     """
     test_pairs = []
-    with open(os.path.dirname(os.path.abspath(__file__)) + os.path.sep + file_name, 'r', encoding = 'utf-8') as f:
+    with open(os.path.dirname(os.path.abspath(__file__)) + os.path.sep + file_name, 'r', encoding='utf-8') as f:
         for line in f:
             components = line.strip("\n").split("~")
             spoken = components[0]

tests/nemo_text_processing/utils.py

@@ -309,10 +309,10 @@ def parse_args():
         )
         from nemo_text_processing.text_normalization.ja.verbalizers.verbalize import VerbalizeFst as TNVerbalizeFst
     elif args.language == 'rw':
+        from nemo_text_processing.text_normalization.ko.verbalizers.verbalize import VerbalizeFst as TNVerbalizeFst
         from nemo_text_processing.text_normalization.rw.taggers.tokenize_and_classify import (
             ClassifyFst as TNClassifyFst,
         )
-        from nemo_text_processing.text_normalization.ko.verbalizers.verbalize import VerbalizeFst as TNVerbalizeFst
     elif args.language == 'ko':
         from nemo_text_processing.text_normalization.ko.taggers.tokenize_and_classify import (
             ClassifyFst as TNClassifyFst,