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

for more information, see https://pre-commit.ci
NVIDIA · Sep 26, 2024 · 6afcc2a · 6afcc2a
1 parent 30e462f
commit 6afcc2a
Show file tree

Hide file tree

Showing 34 changed files with 34 additions and 34 deletions.
diff --git a/nemo_text_processing/inverse_text_normalization/es/taggers/cardinal.py b/nemo_text_processing/inverse_text_normalization/es/taggers/cardinal.py
@@ -205,4 +205,4 @@ def delete_word(self, word: str):
             if len(word) > 0:
                 delete_graph |= pynutil.delete(word[0].upper() + word[1:])
 
-        return delete_graph.optimize()
+        return delete_graph.optimize()
diff --git a/nemo_text_processing/inverse_text_normalization/es/taggers/date.py b/nemo_text_processing/inverse_text_normalization/es/taggers/date.py
@@ -87,4 +87,4 @@ def __init__(self, cardinal: GraphFst, input_case: str = INPUT_LOWER_CASED):
             final_graph |= capitalized_input_graph(final_graph)
 
         final_graph = self.add_tokens(final_graph)
-        self.fst = final_graph.optimize()
+        self.fst = final_graph.optimize()
diff --git a/nemo_text_processing/inverse_text_normalization/es/taggers/electronic.py b/nemo_text_processing/inverse_text_normalization/es/taggers/electronic.py
@@ -170,4 +170,4 @@ def __init__(self, input_case: str = INPUT_LOWER_CASED):
             graph = capitalized_input_graph(graph, capitalized_graph_weight=MIN_POS_WEIGHT)
 
         final_graph = self.add_tokens(graph)
-        self.fst = final_graph.optimize()
+        self.fst = final_graph.optimize()
diff --git a/nemo_text_processing/inverse_text_normalization/es/taggers/telephone.py b/nemo_text_processing/inverse_text_normalization/es/taggers/telephone.py
@@ -164,4 +164,4 @@ def __init__(self, input_case: str = INPUT_LOWER_CASED):
             graph |= capitalized_input_graph(graph)
 
         final_graph = self.add_tokens(graph)
-        self.fst = final_graph.optimize()
+        self.fst = final_graph.optimize()
diff --git a/nemo_text_processing/inverse_text_normalization/ja/taggers/cardinal.py b/nemo_text_processing/inverse_text_normalization/ja/taggers/cardinal.py
@@ -152,4 +152,4 @@ def __init__(self):
         ) | (pynutil.insert("integer: \"") + graph + pynutil.insert("\""))
 
         final_graph = self.add_tokens(final_graph)
-        self.fst = final_graph.optimize()
+        self.fst = final_graph.optimize()
diff --git a/nemo_text_processing/inverse_text_normalization/ja/taggers/decimal.py b/nemo_text_processing/inverse_text_normalization/ja/taggers/decimal.py
@@ -65,4 +65,4 @@ def __init__(self, cardinal: GraphFst):
         )
 
         final_graph = self.add_tokens(final_graph)
-        self.fst = final_graph.optimize()
+        self.fst = final_graph.optimize()
diff --git a/nemo_text_processing/inverse_text_normalization/ja/taggers/fraction.py b/nemo_text_processing/inverse_text_normalization/ja/taggers/fraction.py
@@ -100,4 +100,4 @@ def __init__(self, cardinal: GraphFst, decimal: GraphFst):
         final_graph = graph_regular_fractions | graph_integer_fraction_sign
 
         final_graph = self.add_tokens(final_graph)
-        self.fst = final_graph.optimize()
+        self.fst = final_graph.optimize()
diff --git a/nemo_text_processing/inverse_text_normalization/ja/taggers/ordinal.py b/nemo_text_processing/inverse_text_normalization/ja/taggers/ordinal.py
@@ -37,4 +37,4 @@ def __init__(self, cardinal: GraphFst):
         )
 
         final_graph = self.add_tokens(integer_component)
-        self.fst = final_graph.optimize()
+        self.fst = final_graph.optimize()
diff --git a/nemo_text_processing/inverse_text_normalization/ja/taggers/preprocessor.py b/nemo_text_processing/inverse_text_normalization/ja/taggers/preprocessor.py
@@ -46,4 +46,4 @@ def __init__(
             fullwidth_to_halfwidth_graph = pynini.string_file(get_abs_path('data/char/fullwidth_to_halfwidth.tsv'))
             graph @= pynini.cdrewrite(fullwidth_to_halfwidth_graph, '', '', NEMO_SIGMA)
 
-        self.fst = graph.optimize()
+        self.fst = graph.optimize()
diff --git a/nemo_text_processing/inverse_text_normalization/ja/verbalizers/date.py b/nemo_text_processing/inverse_text_normalization/ja/verbalizers/date.py
@@ -61,4 +61,4 @@ def __init__(self):
         graph = graph_regular | era_component
 
         final_graph = self.delete_tokens(graph)
-        self.fst = final_graph.optimize()
+        self.fst = final_graph.optimize()
diff --git a/nemo_text_processing/inverse_text_normalization/ja/verbalizers/fraction.py b/nemo_text_processing/inverse_text_normalization/ja/verbalizers/fraction.py
@@ -62,4 +62,4 @@ def __init__(self):
 
         final_graph = self.delete_tokens(regular_graph)
 
-        self.fst = final_graph.optimize()
+        self.fst = final_graph.optimize()
diff --git a/nemo_text_processing/inverse_text_normalization/ja/verbalizers/post_processing.py b/nemo_text_processing/inverse_text_normalization/ja/verbalizers/post_processing.py
@@ -129,4 +129,4 @@ def get_punct_postprocess_graph(self):
 
         graph = remove_space_around_single_quote.optimize()
 
-        return graph
+        return graph
diff --git a/nemo_text_processing/inverse_text_normalization/ja/verbalizers/postprocessor.py b/nemo_text_processing/inverse_text_normalization/ja/verbalizers/postprocessor.py
@@ -74,4 +74,4 @@ def __init__(
             tag_oov_graph = pynutil.insert(ltag) + oov_charset + pynutil.insert(rtag)
             graph @= pynini.cdrewrite(tag_oov_graph, "", "", NEMO_SIGMA).optimize()
 
-        self.fst = graph.optimize()
+        self.fst = graph.optimize()
diff --git a/nemo_text_processing/inverse_text_normalization/ja/verbalizers/time.py b/nemo_text_processing/inverse_text_normalization/ja/verbalizers/time.py
@@ -60,4 +60,4 @@ def __init__(self):
         final_graph = graph
 
         final_graph = self.delete_tokens(final_graph.optimize())
-        self.fst = final_graph.optimize()
+        self.fst = final_graph.optimize()
diff --git a/nemo_text_processing/inverse_text_normalization/ja/verbalizers/verbalize.py b/nemo_text_processing/inverse_text_normalization/ja/verbalizers/verbalize.py
@@ -55,4 +55,4 @@ def __init__(self):
         graph = (
             cardinal_graph | date_graph | time_graph | ordinal_graph | decimal_graph | fraction_graph | whitelist_graph
         )
-        self.fst = graph
+        self.fst = graph
diff --git a/nemo_text_processing/inverse_text_normalization/ja/verbalizers/verbalize_final.py b/nemo_text_processing/inverse_text_normalization/ja/verbalizers/verbalize_final.py
@@ -51,4 +51,4 @@ def __init__(self, deterministic: bool = True, cache_dir: str = None, overwrite_
 
             self.fst = (verbalizer @ postprocessor.fst).optimize()
             if far_file:
-                generator_main(far_file, {"verbalize": self.fst})
+                generator_main(far_file, {"verbalize": self.fst})
diff --git a/nemo_text_processing/inverse_text_normalization/mr/taggers/decimal.py b/nemo_text_processing/inverse_text_normalization/mr/taggers/decimal.py
@@ -95,4 +95,4 @@ def __init__(self, cardinal: GraphFst):
         final_graph |= optional_graph_negative + quantity_graph
 
         final_graph = self.add_tokens(final_graph)
-        self.fst = final_graph.optimize()
+        self.fst = final_graph.optimize()
diff --git a/nemo_text_processing/inverse_text_normalization/mr/taggers/time.py b/nemo_text_processing/inverse_text_normalization/mr/taggers/time.py
@@ -114,4 +114,4 @@ def __init__(self):
 
         final_graph = graph
         final_graph = self.add_tokens(final_graph)
-        self.fst = final_graph.optimize()
+        self.fst = final_graph.optimize()
diff --git a/nemo_text_processing/inverse_text_normalization/mr/utils.py b/nemo_text_processing/inverse_text_normalization/mr/utils.py
@@ -57,4 +57,4 @@ def augment_labels_with_punct_at_end(labels):
         if len(label) > 1:
             if label[0][-1] == "." and label[1][-1] != ".":
                 res.append([label[0], label[1] + "."] + label[2:])
-    return res
+    return res
diff --git a/nemo_text_processing/inverse_text_normalization/mr/verbalizers/time.py b/nemo_text_processing/inverse_text_normalization/mr/verbalizers/time.py
@@ -52,4 +52,4 @@ def __init__(self):
         )
 
         delete_tokens = self.delete_tokens(graph)
-        self.fst = delete_tokens.optimize()
+        self.fst = delete_tokens.optimize()
diff --git a/nemo_text_processing/text_normalization/en/graph_utils.py b/nemo_text_processing/text_normalization/en/graph_utils.py
@@ -304,4 +304,4 @@ def delete_tokens(self, fst) -> "pynini.FstLike":
             + delete_space
             + pynutil.delete("}")
         )
-        return res @ pynini.cdrewrite(pynini.cross("\u00A0", " "), "", "", NEMO_SIGMA)
+        return res @ pynini.cdrewrite(pynini.cross("\u00A0", " "), "", "", NEMO_SIGMA)
diff --git a/nemo_text_processing/text_normalization/en/taggers/electronic.py b/nemo_text_processing/text_normalization/en/taggers/electronic.py
@@ -165,4 +165,4 @@ def __init__(self, cardinal: GraphFst, deterministic: bool = True):
 
         final_graph = self.add_tokens(graph)
 
-        self.fst = final_graph.optimize()
+        self.fst = final_graph.optimize()
diff --git a/nemo_text_processing/text_normalization/en/taggers/tokenize_and_classify.py b/nemo_text_processing/text_normalization/en/taggers/tokenize_and_classify.py
@@ -229,4 +229,4 @@ def __init__(
             self.fst = graph.optimize()
 
             if far_file:
-                generator_main(far_file, {"tokenize_and_classify": self.fst})
+                generator_main(far_file, {"tokenize_and_classify": self.fst})
diff --git a/nemo_text_processing/text_normalization/es/graph_utils.py b/nemo_text_processing/text_normalization/es/graph_utils.py
@@ -212,4 +212,4 @@ def _load_roman(file: str, upper_casing: bool):
         )
     ).optimize()
 
-    return graph @ fst
+    return graph @ fst
diff --git a/nemo_text_processing/text_normalization/fr/verbalizers/__init__.py b/nemo_text_processing/text_normalization/fr/verbalizers/__init__.py
@@ -10,4 +10,4 @@
 # distributed under the License is distributed on an "AS IS" BASIS,
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
-# limitations under the License.
+# limitations under the License.
diff --git a/nemo_text_processing/text_normalization/it/taggers/electronic.py b/nemo_text_processing/text_normalization/it/taggers/electronic.py
@@ -104,4 +104,4 @@ def __init__(self, deterministic: bool = True):
         self.graph = graph
 
         final_graph = self.add_tokens(self.graph + pynutil.insert(" preserve_order: true"))
-        self.fst = final_graph.optimize()
+        self.fst = final_graph.optimize()
diff --git a/nemo_text_processing/text_normalization/it/taggers/tokenize_and_classify.py b/nemo_text_processing/text_normalization/it/taggers/tokenize_and_classify.py
@@ -126,4 +126,4 @@ def __init__(
             self.fst_no_digits = pynini.compose(self.fst, no_digits).optimize()
 
             if far_file:
-                generator_main(far_file, {"tokenize_and_classify": self.fst})
+                generator_main(far_file, {"tokenize_and_classify": self.fst})
diff --git a/nemo_text_processing/text_normalization/zh/taggers/money.py b/nemo_text_processing/text_normalization/zh/taggers/money.py
@@ -136,4 +136,4 @@ def __init__(self, cardinal: GraphFst, deterministic: bool = True, lm: bool = Fa
         )
 
         final_graph = self.add_tokens(graph)
-        self.fst = final_graph.optimize()
+        self.fst = final_graph.optimize()
diff --git a/nemo_text_processing/text_normalization/zh/taggers/tokenize_and_classify.py b/nemo_text_processing/text_normalization/zh/taggers/tokenize_and_classify.py
@@ -97,4 +97,4 @@ def __init__(
             self.fst = tagger
 
             if far_file:
-                generator_main(far_file, {"tokenize_and_classify": self.fst})
+                generator_main(far_file, {"tokenize_and_classify": self.fst})
diff --git a/nemo_text_processing/text_normalization/zh/verbalizers/word.py b/nemo_text_processing/text_normalization/zh/verbalizers/word.py
@@ -27,4 +27,4 @@ def __init__(self, deterministic: bool = True, lm: bool = False):
         super().__init__(name="char", kind="verbalize", deterministic=deterministic)
 
         graph = pynutil.delete("name: \"") + NEMO_NOT_QUOTE + pynutil.delete("\"")
-        self.fst = graph.optimize()
+        self.fst = graph.optimize()
diff --git a/requirements/requirements.txt b/requirements/requirements.txt
@@ -10,4 +10,4 @@ setuptools>=65.5.1
 tqdm>=4.41.0
 transformers
 wget
-wrapt
+wrapt
diff --git a/requirements/requirements_test.txt b/requirements/requirements_test.txt
@@ -10,4 +10,4 @@ sphinx
 sphinxcontrib-bibtex
 wandb
 wget
-wrapt
+wrapt
diff --git a/tests/nemo_text_processing/mr/test_date.py b/tests/nemo_text_processing/mr/test_date.py
@@ -29,4 +29,4 @@ class TestDate:
     @pytest.mark.unit
     def test_denorm(self, test_input, expected):
         pred = self.inverse_normalizer_mr.inverse_normalize(test_input, verbose=False)
-        assert pred == expected
+        assert pred == expected
diff --git a/tools/text_processing_deployment/pynini_export.py b/tools/text_processing_deployment/pynini_export.py
@@ -288,10 +288,10 @@ def parse_args():
         from nemo_text_processing.text_normalization.ja.taggers.tokenize_and_classify import (
             ClassifyFst as TNClassifyFst,
         )
-        from nemo_text_processing.text_normalization.ja.verbalizers.verbalize import VerbalizeFst as TNVerbalizeFst
         from nemo_text_processing.text_normalization.ja.verbalizers.post_processing import (
             PostProcessingFst as TNPostProcessingFst,
         )
+        from nemo_text_processing.text_normalization.ja.verbalizers.verbalize import VerbalizeFst as TNVerbalizeFst
     elif args.language == 'rw':
         from nemo_text_processing.text_normalization.rw.taggers.tokenize_and_classify import (
             ClassifyFst as TNClassifyFst,