zhifu gao
2024-04-24 861147c7308b91068ffa02724fdf74ee623a909e
fun_text_processing/text_normalization/es/verbalizers/cardinal.py
@@ -1,5 +1,3 @@
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 (
@@ -22,11 +20,11 @@
    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