wuhongsheng
2024-07-05 3a4281f4959534b1bf5d01acf0085f4f8e6f2ec8
fun_text_processing/text_normalization/zh/verbalizers/cardinal.py
@@ -4,13 +4,13 @@
class Cardinal(GraphFst):
    '''
        tokens { cardinal { integer: "一二三" } } -> 一二三
    '''
    """
    tokens { cardinal { integer: "一二三" } } -> 一二三
    """
    def __init__(self, deterministic: bool = True, lm: bool = False):
        super().__init__(name="cardinal", kind="verbalize", deterministic=deterministic)
        graph = pynutil.delete('integer: \"') + pynini.closure(FUN_NOT_QUOTE) + pynutil.delete('\"')
        graph = pynutil.delete('integer: "') + pynini.closure(FUN_NOT_QUOTE) + pynutil.delete('"')
        self.fst = self.delete_tokens(graph).optimize()