From b64fb60a0d888c04b039d73b393ce7f498b1c9e9 Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Mon, 10 Apr 2023 12:34:34 +0800 Subject: [PATCH] small fixes --- icefall/shared/combine_lm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/icefall/shared/combine_lm.py b/icefall/shared/combine_lm.py index 8e117a4ae..9c912006f 100755 --- a/icefall/shared/combine_lm.py +++ b/icefall/shared/combine_lm.py @@ -179,7 +179,7 @@ def _process_grams( def process(args): b = kenlm.LanguageModel(args.b) logging.info(f"Order of {args.b}: {b.order}") - pattern = re.compile(r"\\(\d+)-grams:") + pattern = re.compile(r"\\(\d+)-grams:\n") out = open(args.out, "w", encoding="utf-8") a_scale = args.a_scale @@ -188,7 +188,7 @@ def process(args): with open(args.a, encoding="utf-8") as a: for line in a: print(line, end="", file=out) - m = pattern.search(line) + m = pattern.match(line) if m: order = int(m.group(1)) _process_grams(