From 362c9d37a9e82b861d794ffc9dd11cd43d066cce Mon Sep 17 00:00:00 2001 From: quillotm Date: Thu, 2 Sep 2021 17:06:30 +0200 Subject: [PATCH] Fixed simple issue --- volia/masseffect.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/volia/masseffect.py b/volia/masseffect.py index 0163eea..c3c4615 100644 --- a/volia/masseffect.py +++ b/volia/masseffect.py @@ -99,8 +99,9 @@ def utt2sub(file: str, outfile: str): with open(outfile, "w") as of: key_2_subkeys = {} for subkey in keys: - key = subkey.replace(" ", "").replace("\n", "").split("_")[:-1] - key_2_subkeys[key] = [] + key = "_".join(subkey.replace(" ", "").replace("\n", "").split("_")[:-1]) + if key not in key_2_subkeys: + key_2_subkeys[key] = [] key_2_subkeys[key].append(subkey) for key in key_2_subkeys: @@ -114,7 +115,7 @@ def sub2utt(file: str, outfile: str): with open(outfile, "w") as of: for subkey in keys: - key = subkey.replace(" ", "").replace("\n", "").split("_")[:-1] + key = "_".join(subkey.replace(" ", "").replace("\n", "").split("_")[:-1]) of.write(f"{subkey} {key}\n") -- 1.8.2.3