From b75d1e89bb2f513a79bb07e9100ba1cd2bbcf40c Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期日, 09 六月 2024 00:32:57 +0800
Subject: [PATCH] fix bug
---
fun_text_processing/inverse_text_normalization/zh/verbalizers/time.py | 27 ++++++++++++++-------------
1 files changed, 14 insertions(+), 13 deletions(-)
diff --git a/fun_text_processing/inverse_text_normalization/zh/verbalizers/time.py b/fun_text_processing/inverse_text_normalization/zh/verbalizers/time.py
index 60a0d5e..2fdc919 100755
--- a/fun_text_processing/inverse_text_normalization/zh/verbalizers/time.py
+++ b/fun_text_processing/inverse_text_normalization/zh/verbalizers/time.py
@@ -1,4 +1,3 @@
-
import pynini
from fun_text_processing.inverse_text_normalization.zh.graph_utils import (
DAMO_CHAR,
@@ -20,29 +19,31 @@
def __init__(self):
super().__init__(name="time", kind="verbalize")
- add_leading_zero_to_double_digit = (DAMO_DIGIT + DAMO_DIGIT) | (pynutil.insert("0") + DAMO_DIGIT)
-
+ add_leading_zero_to_double_digit = (DAMO_DIGIT + DAMO_DIGIT) | (
+ pynutil.insert("0") + DAMO_DIGIT
+ )
+
hour = (
pynutil.delete("hour:")
+ delete_space
- + pynutil.delete("\"")
+ + pynutil.delete('"')
+ pynini.closure(DAMO_DIGIT, 1)
- + pynutil.delete("\"")
+ + pynutil.delete('"')
)
minute = (
pynutil.delete("minute:")
+ delete_space
- + pynutil.delete("\"")
+ + pynutil.delete('"')
+ pynini.closure(DAMO_DIGIT, 1)
- + pynutil.delete("\"")
+ + pynutil.delete('"')
)
second = (
pynutil.delete("second:")
+ delete_space
- + pynutil.delete("\"")
+ + pynutil.delete('"')
+ pynini.closure(DAMO_DIGIT, 1)
- + pynutil.delete("\"")
+ + pynutil.delete('"')
)
suffix = (
@@ -50,9 +51,9 @@
+ pynutil.insert(" ")
+ pynutil.delete("noon:")
+ delete_space
- + pynutil.delete("\"")
+ + pynutil.delete('"')
+ pynini.closure(DAMO_CHAR - " ", 1)
- + pynutil.delete("\"")
+ + pynutil.delete('"')
)
optional_suffix = pynini.closure(suffix, 0, 1)
zone = (
@@ -60,9 +61,9 @@
+ pynutil.insert(" ")
+ pynutil.delete("zone:")
+ delete_space
- + pynutil.delete("\"")
+ + pynutil.delete('"')
+ pynini.closure(DAMO_CHAR - " ", 1)
- + pynutil.delete("\"")
+ + pynutil.delete('"')
)
optional_zone = pynini.closure(zone, 0, 1)
graph = (
--
Gitblit v1.9.1