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

for more information, see https://pre-commit.ci
NVIDIA · Jul 18, 2023 · 29cd7f6 · 29cd7f6
1 parent d6b8ae9
commit 29cd7f6
Show file tree

Hide file tree

Showing 11 changed files with 15 additions and 14 deletions.
diff --git a/nemo_text_processing/text_normalization/fr/taggers/date.py b/nemo_text_processing/text_normalization/fr/taggers/date.py
@@ -20,7 +20,6 @@
 from pynini.lib import pynutil
 
 
-
 class DateFst(GraphFst):
     # MAKE SURE ANY COMMENTS APPLY TO YOUR LANGUAGE
     """

diff --git a/nemo_text_processing/text_normalization/fr/taggers/decimals.py b/nemo_text_processing/text_normalization/fr/taggers/decimals.py
@@ -15,17 +15,16 @@
 
 import pynini
 from nemo_text_processing.text_normalization.en.graph_utils import (
-    delete_space,
-    insert_space,
     NEMO_DIGIT,
     NEMO_SIGMA,
     NEMO_SPACE,
     GraphFst,
+    delete_space,
+    insert_space,
 )
 from nemo_text_processing.text_normalization.fr.utils import get_abs_path
 from pynini.lib import pynutil
 
-
 quantities = pynini.string_file(get_abs_path("data/numbers/quantities.tsv"))
 digit = pynini.string_file(get_abs_path("data/numbers/digit.tsv"))
 zero = pynini.string_file(get_abs_path("data/numbers/zero.tsv"))

diff --git a/nemo_text_processing/text_normalization/fr/taggers/electronic.py b/nemo_text_processing/text_normalization/fr/taggers/electronic.py
@@ -41,4 +41,3 @@ def __init__(self, deterministic: bool = True):
 
         final_graph = self.add_tokens(final_graph)
         self.fst = final_graph.optimize()
-
diff --git a/nemo_text_processing/text_normalization/fr/taggers/money.py b/nemo_text_processing/text_normalization/fr/taggers/money.py
@@ -48,4 +48,4 @@ def __init__(self, cardinal: GraphFst, decimal: GraphFst, deterministic: bool =
         final_graph = pynutil.insert("integer_part: \"") + pynini.closure(NEMO_NOT_SPACE, 1) + 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/text_normalization/fr/verbalizers/decimals.py b/nemo_text_processing/text_normalization/fr/verbalizers/decimals.py
@@ -13,11 +13,11 @@
 # limitations under the License.
 import pynini
 from nemo_text_processing.text_normalization.en.graph_utils import (
+    NEMO_NOT_QUOTE,
+    GraphFst,
     delete_preserve_order,
     delete_space,
     insert_space,
-    NEMO_NOT_QUOTE,
-    GraphFst,
 )
 from pynini.lib import pynutil
 

diff --git a/nemo_text_processing/text_normalization/fr/verbalizers/electronic.py b/nemo_text_processing/text_normalization/fr/verbalizers/electronic.py
@@ -36,4 +36,4 @@ def __init__(self, deterministic: bool = True):
         # A VARIABLE CALLED final_graph WITH AN FST COMPRISED OF ALL THE RULES
         final_graph = pynutil.delete("username: \"") + pynini.closure(NEMO_NOT_QUOTE, 1) + pynutil.delete("\"")
 
-        self.fst = self.delete_tokens(final_graph).optimize()
+        self.fst = self.delete_tokens(final_graph).optimize()
diff --git a/nemo_text_processing/text_normalization/fr/verbalizers/fraction.py b/nemo_text_processing/text_normalization/fr/verbalizers/fraction.py
@@ -13,11 +13,11 @@
 # limitations under the License.
 import pynini
 from nemo_text_processing.text_normalization.en.graph_utils import (
-    insert_space,
     NEMO_NOT_QUOTE,
     NEMO_SIGMA,
     NEMO_SPACE,
     GraphFst,
+    insert_space,
 )
 from nemo_text_processing.text_normalization.fr.utils import get_abs_path
 from pynini.lib import pynutil

diff --git a/nemo_text_processing/text_normalization/normalize.py b/nemo_text_processing/text_normalization/normalize.py
@@ -689,7 +689,11 @@ def parse_args():
     )
     parser.add_argument('--output_file', dest="output_file", help="Output file path", type=str)
     parser.add_argument(
-        "--language", help="language", choices=["en", "de", "es", "fr", "hu", "sv", "zh", "ar", "it"], default="en", type=str
+        "--language",
+        help="language",
+        choices=["en", "de", "es", "fr", "hu", "sv", "zh", "ar", "it"],
+        default="en",
+        type=str,
     )
     parser.add_argument(
         "--input_case",

diff --git a/tests/nemo_text_processing/fr/test_cardinal.py b/tests/nemo_text_processing/fr/test_cardinal.py
@@ -38,4 +38,4 @@ def test_denorm(self, test_input, expected):
     @pytest.mark.unit
     def test_norm(self, test_input, expected):
         pred = self.normalizer.normalize(test_input, verbose=False)
-        assert pred in expected
+        assert pred in expected
diff --git a/tests/nemo_text_processing/fr/test_decimal.py b/tests/nemo_text_processing/fr/test_decimal.py
@@ -37,4 +37,4 @@ def test_denorm(self, test_input, expected):
     @pytest.mark.unit
     def test_norm(self, test_input, expected):
         pred = self.normalizer.normalize(test_input, verbose=False)
-        assert pred == expected
+        assert pred == expected
diff --git a/tests/nemo_text_processing/fr/test_fraction.py b/tests/nemo_text_processing/fr/test_fraction.py
@@ -38,4 +38,4 @@ def test_denorm(self, test_input, expected):
     @pytest.mark.unit
     def test_norm(self, test_input, expected):
         pred = self.normalizer.normalize(test_input, verbose=False)
-        assert pred == expected
+        assert pred == expected