Skip to content

Commit bc713c6

Browse files
committed
incorporated the suggestion for decimal graph
Signed-off-by: Namrata Gachchi <ngachchi@nvidia.com>
1 parent 5e6e65a commit bc713c6

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

nemo_text_processing/text_normalization/hi/taggers/cardinal.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -80,6 +80,7 @@ def create_larger_number_graph(digit_graph, suffix, zeros_counts, sub_graph):
8080
graph_ten_thousands |= create_larger_number_graph(teens_and_ties, suffix_thousands, 1, teens_ties)
8181
graph_ten_thousands |= create_larger_number_graph(teens_and_ties, suffix_thousands, 0, graph_hundreds)
8282
graph_ten_thousands.optimize()
83+
self.graph_ten_thousands = graph_ten_thousands
8384

8485
# Lakhs graph and ten lakhs graph
8586
suffix_lakhs = pynutil.insert(" लाख")
@@ -90,6 +91,7 @@ def create_larger_number_graph(digit_graph, suffix, zeros_counts, sub_graph):
9091
graph_lakhs |= create_larger_number_graph(digit, suffix_lakhs, 1, graph_thousands)
9192
graph_lakhs |= create_larger_number_graph(digit, suffix_lakhs, 0, graph_ten_thousands)
9293
graph_lakhs.optimize()
94+
self.graph_lakhs = graph_lakhs
9395

9496
graph_ten_lakhs = create_graph_suffix(teens_and_ties, suffix_lakhs, 5)
9597
graph_ten_lakhs |= create_larger_number_graph(teens_and_ties, suffix_lakhs, 4, digit)
@@ -98,6 +100,7 @@ def create_larger_number_graph(digit_graph, suffix, zeros_counts, sub_graph):
98100
graph_ten_lakhs |= create_larger_number_graph(teens_and_ties, suffix_lakhs, 1, graph_thousands)
99101
graph_ten_lakhs |= create_larger_number_graph(teens_and_ties, suffix_lakhs, 0, graph_ten_thousands)
100102
graph_ten_lakhs.optimize()
103+
self.graph_ten_lakhs = graph_ten_lakhs
101104

102105
# Crores graph ten crores graph
103106
suffix_crores = pynutil.insert(" करोड़")

nemo_text_processing/text_normalization/hi/taggers/measure.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,9 +40,10 @@ class MeasureFst(GraphFst):
4040
def __init__(self, cardinal: GraphFst, decimal: GraphFst):
4141
super().__init__(name="measure", kind="classify")
4242

43-
cardinal_graph = digit | teens_and_ties | cardinal.graph_hundreds | cardinal.graph_thousands
43+
cardinal_graph = digit | teens_and_ties | cardinal.graph_hundreds | cardinal.graph_thousands | cardinal.graph_ten_thousands | cardinal.graph_lakhs | cardinal.graph_ten_lakhs
4444
point = pynutil.delete(".")
45-
decimal_graph = decimal.graph_integer + point + insert_space + decimal.graph_fractional
45+
decimal_integers = pynutil.insert("integer_part: \"") + cardinal_graph + pynutil.insert("\"")
46+
decimal_graph = decimal_integers + point + insert_space + decimal.graph_fractional
4647
unit_graph = pynini.string_file(get_abs_path("data/measure/unit.tsv"))
4748

4849
optional_graph_negative = pynini.closure(

0 commit comments

Comments
 (0)