From 810046e3df4910c8f5c1a90e4b53aca45b3397e8 Mon Sep 17 00:00:00 2001
From: wuhongsheng <664116298@qq.com>
Date: 星期一, 01 七月 2024 10:42:58 +0800
Subject: [PATCH] 优化merge segments 参数,解决新闻联播男女主持人“晚上好”合并一个speakid问题 (#1861)

---
 fun_text_processing/inverse_text_normalization/zh/verbalizers/whitelist.py |   15 +++++++++------
 1 files changed, 9 insertions(+), 6 deletions(-)

diff --git a/fun_text_processing/inverse_text_normalization/zh/verbalizers/whitelist.py b/fun_text_processing/inverse_text_normalization/zh/verbalizers/whitelist.py
index 42fca61..902b373 100755
--- a/fun_text_processing/inverse_text_normalization/zh/verbalizers/whitelist.py
+++ b/fun_text_processing/inverse_text_normalization/zh/verbalizers/whitelist.py
@@ -1,7 +1,10 @@
-
-
 import pynini
-from fun_text_processing.inverse_text_normalization.zh.graph_utils import DAMO_CHAR, DAMO_SIGMA, GraphFst, delete_space
+from fun_text_processing.inverse_text_normalization.zh.graph_utils import (
+    DAMO_CHAR,
+    DAMO_SIGMA,
+    GraphFst,
+    delete_space,
+)
 from pynini.lib import pynutil
 
 
@@ -16,9 +19,9 @@
         graph = (
             pynutil.delete("name:")
             + delete_space
-            + pynutil.delete("\"")
+            + pynutil.delete('"')
             + pynini.closure(DAMO_CHAR - " ", 1)
-            + pynutil.delete("\"")
+            + pynutil.delete('"')
         )
-        graph = graph @ pynini.cdrewrite(pynini.cross(u"\u00A0", " "), "", "", DAMO_SIGMA)
+        graph = graph @ pynini.cdrewrite(pynini.cross("\u00A0", " "), "", "", DAMO_SIGMA)
         self.fst = graph.optimize()

--
Gitblit v1.9.1