mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-08 09:32:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix str2bool (#1792)

Browse Source
This commit is contained in:
Yifan Yang 2024-10-31 09:54:12 +08:00 committed by GitHub
parent 87cadfcd2e
commit 119e1ce3e8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/libriheavy/ASR/local/prepare_manifest.py
View File

@ -20,6 +20,8 @@ import json
import sys import sys
from pathlib import Path from pathlib import Path
from icefall.utils import str2bool
def simple_cleanup(text: str) -> str: def simple_cleanup(text: str) -> str:
table = str.maketrans("’‘，。；？！（）：-《》、“”【】", "'',.;?!(): <>/\"\"[]") table = str.maketrans("’‘，。；？！（）：-《》、“”【】", "'',.;?!(): <>/\"\"[]")
@ -34,7 +36,7 @@ def main():
), "Usage: ./local/prepare_manifest.py INPUT OUTPUT_DIR KEEP_CUSTOM_FIELDS" ), "Usage: ./local/prepare_manifest.py INPUT OUTPUT_DIR KEEP_CUSTOM_FIELDS"
fname = Path(sys.argv[1]).name fname = Path(sys.argv[1]).name
oname = Path(sys.argv[2]) / fname oname = Path(sys.argv[2]) / fname
keep_custom_fields = bool(sys.argv[3]) keep_custom_fields = str2bool(sys.argv[3])
with gzip.open(sys.argv[1], "r") as fin, gzip.open(oname, "w") as fout: with gzip.open(sys.argv[1], "r") as fin, gzip.open(oname, "w") as fout:
for line in fin: for line in fin:
cut = json.loads(line) cut = json.loads(line)