[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/vi/__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.

nemo_text_processing/text_normalization/vi/taggers/__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.

nemo_text_processing/text_normalization/vi/taggers/cardinal.py

@@ -54,4 +54,4 @@ def __init__(self, deterministic: bool = True):
         graph = pynutil.insert('name: "') + punct + pynutil.insert('"')
 
         final_graph = pynutil.insert("punctuation { ") + graph + pynutil.insert(" }")
-        self.fst = final_graph.optimize()
+        self.fst = final_graph.optimize()

nemo_text_processing/text_normalization/vi/taggers/punctuation.py

@@ -46,7 +46,10 @@ def __init__(
         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"vi_tn_{deterministic}_deterministic_{input_case}_tokenize.far",)
+            far_file = os.path.join(
+                cache_dir,
+                f"vi_tn_{deterministic}_deterministic_{input_case}_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}.")
@@ -73,9 +76,9 @@ def __init__(
             logger.debug(f"word: {time.time() - start_time: .2f}s -- {word_graph.num_states()} nodes")
 
             classify = (
-                pynutil.add_weight(whitelist_graph, 0.8)  
-                | pynutil.add_weight(cardinal_graph, 0.9) 
-                | pynutil.add_weight(word_graph, 100)  
+                pynutil.add_weight(whitelist_graph, 0.8)
+                | pynutil.add_weight(cardinal_graph, 0.9)
+                | pynutil.add_weight(word_graph, 100)
             )
             punct = pynutil.insert("tokens { ") + pynutil.add_weight(punct_graph, weight=2.1) + pynutil.insert(" }")
             token = pynutil.insert("tokens { ") + classify + pynutil.insert(" }")
@@ -89,4 +92,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})

nemo_text_processing/text_normalization/vi/taggers/tokenize_and_classify.py

@@ -15,8 +15,8 @@
 import pynini
 from pynini.lib import pynutil
 
-from nemo_text_processing.text_normalization.vi.utils import get_abs_path, load_labels
 from nemo_text_processing.text_normalization.en.graph_utils import GraphFst, convert_space
+from nemo_text_processing.text_normalization.vi.utils import get_abs_path, load_labels
 
 
 class WhiteListFst(GraphFst):
@@ -67,4 +67,4 @@ def _get_whitelist_graph(input_case, file):
         self.fst = (pynutil.insert("name: \"") + self.final_graph + pynutil.insert("\"")).optimize()
 
         # Add tokens wrapper
-        self.fst = self.add_tokens(self.fst) 
+        self.fst = self.add_tokens(self.fst)

nemo_text_processing/text_normalization/vi/taggers/whitelist.py

@@ -31,4 +31,4 @@ class WordFst(GraphFst):
     def __init__(self, deterministic: bool = True):
         super().__init__(name="word", kind="classify", deterministic=deterministic)
         word = pynutil.insert("name: \"") + pynini.closure(NEMO_NOT_SPACE, 1) + pynutil.insert("\"")
-        self.fst = word.optimize()
+        self.fst = word.optimize()

nemo_text_processing/text_normalization/vi/taggers/word.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.

nemo_text_processing/text_normalization/vi/verbalizers/__init__.py

@@ -52,4 +52,4 @@ def __init__(self, deterministic: bool = True):
 
         # Delete the token structure and create final FST
         delete_tokens = self.delete_tokens(self.numbers)
-        self.fst = delete_tokens.optimize()
+        self.fst = delete_tokens.optimize()

nemo_text_processing/text_normalization/vi/verbalizers/cardinal.py

@@ -25,7 +25,7 @@ def __init__(self, deterministic: bool = True):
         # Initialize verbalizers
         cardinal = CardinalFst(deterministic=deterministic)
         cardinal_graph = cardinal.fst
-        
+
         whitelist = WhiteListFst(deterministic=deterministic)
         whitelist_graph = whitelist.fst
 
@@ -35,4 +35,4 @@ def __init__(self, deterministic: bool = True):
         # Combine all verbalizers
         graph = cardinal_graph | whitelist_graph | word_graph
 
-        self.fst = graph
+        self.fst = graph