| | |
| | | |
| | | import pynini |
| | | from fun_text_processing.inverse_text_normalization.fr.graph_utils import ( |
| | | DAMO_NOT_QUOTE, |
| | |
| | | |
| | | def __init__(self): |
| | | super().__init__(name="fraction", kind="verbalize") |
| | | optional_sign = pynini.closure(pynini.cross("negative: \"true\"", "-") + delete_space, 0, 1) |
| | | optional_sign = pynini.closure(pynini.cross('negative: "true"', "-") + delete_space, 0, 1) |
| | | integer = ( |
| | | pynutil.delete("integer_part: \"") |
| | | pynutil.delete('integer_part: "') |
| | | + pynini.closure(DAMO_NOT_QUOTE, 1) |
| | | + pynutil.delete("\"") |
| | | + pynutil.delete('"') |
| | | + insert_space |
| | | ) |
| | | numerator = pynutil.delete("numerator: \"") + pynini.closure(DAMO_NOT_QUOTE, 1) + pynutil.delete("\"") |
| | | |
| | | denominator = ( |
| | | pynutil.insert('/') |
| | | + pynutil.delete("denominator: \"") |
| | | + pynini.closure(DAMO_NOT_QUOTE, 1) |
| | | + pynutil.delete("\"") |
| | | numerator = ( |
| | | pynutil.delete('numerator: "') + pynini.closure(DAMO_NOT_QUOTE, 1) + pynutil.delete('"') |
| | | ) |
| | | |
| | | graph = (pynini.closure(integer + delete_space, 0, 1) + numerator + delete_space + denominator).optimize() |
| | | denominator = ( |
| | | pynutil.insert("/") |
| | | + pynutil.delete('denominator: "') |
| | | + pynini.closure(DAMO_NOT_QUOTE, 1) |
| | | + pynutil.delete('"') |
| | | ) |
| | | |
| | | graph = ( |
| | | pynini.closure(integer + delete_space, 0, 1) + numerator + delete_space + denominator |
| | | ).optimize() |
| | | self.numbers = graph |
| | | delete_tokens = self.delete_tokens(optional_sign + graph) |
| | | self.fst = delete_tokens.optimize() |