| | |
|---|
| | | # 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, GraphFst |
|---|
| | | from fun_text_processing.text_normalization.es.graph_utils import ( |
|---|
| | |
|---|
| | | |
|---|
| | | class CardinalFst(GraphFst): |
|---|
| | | """ |
|---|
| | | Finite state transducer for verbalizing cardinals |
|---|
| | | e.g. cardinal { integer: "dos" } -> "dos" |
|---|
| | | Finite state transducer for verbalizing cardinals |
|---|
| | | e.g. cardinal { integer: "dos" } -> "dos" |
|---|
| | | |
|---|
| | | Args: |
|---|
| | | deterministic: if True will provide a single transduction option, |
|---|
| | | for False multiple transduction are generated (used for audio-based normalization) |
|---|
| | | """ |
|---|
| | | Args: |
|---|
| | | deterministic: if True will provide a single transduction option, |
|---|
| | | for False multiple transduction are generated (used for audio-based normalization) |
|---|
| | | """ |
|---|
| | | |
|---|
| | | def __init__(self, deterministic: bool = True): |
|---|
| | | super().__init__(name="cardinal", kind="verbalize", deterministic=deterministic) |
|---|
| | | optional_sign = pynini.closure(pynini.cross("negative: \"true\" ", "menos "), 0, 1) |
|---|
| | | optional_sign = pynini.closure(pynini.cross('negative: "true" ', "menos "), 0, 1) |
|---|
| | | self.optional_sign = optional_sign |
|---|
| | | |
|---|
| | | integer = pynini.closure(DAMO_NOT_QUOTE, 1) |
|---|
| | | self.integer = pynutil.delete(" \"") + integer + pynutil.delete("\"") |
|---|
| | | self.integer = pynutil.delete(' "') + integer + pynutil.delete('"') |
|---|
| | | |
|---|
| | | integer = pynutil.delete("integer:") + self.integer |
|---|
| | | |
|---|
