liugz18
2024-07-18 d80ac2fd2df4e7fb8a28acfa512bb11472b5cc99
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()