| | |
| | | # Copyright NeMo (https://github.com/NVIDIA/NeMo). All Rights Reserved. |
| | | # |
| | | # Licensed under the Apache License, Version 2.0 (the "License"); |
| | | # you may not use this file except in compliance with the License. |
| | | # You may obtain a copy of the License at |
| | | # |
| | | # http://www.apache.org/licenses/LICENSE-2.0 |
| | | # |
| | | # Unless required by applicable law or agreed to in writing, software |
| | | # 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. |
| | | |
| | | import pynini |
| | | from fun_text_processing.text_normalization.en.graph_utils import ( |
| | | DAMO_NOT_QUOTE, |
| | |
| | | def __init__(self, decimal: GraphFst, deterministic: bool = True): |
| | | super().__init__(name="money", kind="verbalize", deterministic=deterministic) |
| | | keep_space = pynini.accep(" ") |
| | | maj = pynutil.delete("currency_maj: \"") + pynini.closure(DAMO_NOT_QUOTE, 1) + pynutil.delete("\"") |
| | | min = pynutil.delete("currency_min: \"") + pynini.closure(DAMO_NOT_QUOTE, 1) + pynutil.delete("\"") |
| | | maj = ( |
| | | pynutil.delete('currency_maj: "') |
| | | + pynini.closure(DAMO_NOT_QUOTE, 1) |
| | | + pynutil.delete('"') |
| | | ) |
| | | min = ( |
| | | pynutil.delete('currency_min: "') |
| | | + pynini.closure(DAMO_NOT_QUOTE, 1) |
| | | + pynutil.delete('"') |
| | | ) |
| | | |
| | | fractional_part = ( |
| | | pynutil.delete("fractional_part: \"") + pynini.closure(DAMO_NOT_QUOTE, 1) + pynutil.delete("\"") |
| | | pynutil.delete('fractional_part: "') |
| | | + pynini.closure(DAMO_NOT_QUOTE, 1) |
| | | + pynutil.delete('"') |
| | | ) |
| | | |
| | | integer_part = decimal.integer |
| | |
| | | if not deterministic: |
| | | fractional |= pynutil.insert("and ") + fractional |
| | | |
| | | graph_integer_with_minor = integer_part + keep_space + maj + keep_space + fractional + delete_preserve_order |
| | | graph_integer_with_minor = ( |
| | | integer_part + keep_space + maj + keep_space + fractional + delete_preserve_order |
| | | ) |
| | | |
| | | # *** point *** currency_maj |
| | | graph_decimal = decimal.numbers + keep_space + maj |