diff --git a/volia/masseffect.py b/volia/masseffect.py index b5f5ef7..3cbb813 100644 --- a/volia/masseffect.py +++ b/volia/masseffect.py @@ -23,7 +23,7 @@ def utt2char(features: str, outfile: str): splited = key.replace("\n", "").split(",") character = splited[1] f.write(",".join(splited) + " " + character + "\n") - + def char2utt(features: str, outfile: str): raise Exception("Not implemented yet") @@ -74,8 +74,8 @@ def changelabels(source: str, labels: str, outfile: str): with open(outfile, "w") as f: for key in keys: splited = key.split(",") - splited[1] = labels_dict[key] - core.data.write_line(",".join(splited), data_dict[key]) + splited[1] = labels_dict[key][0] + core.data.write_line(",".join(splited), data_dict[key], out=f) if __name__ == '__main__': @@ -119,4 +119,4 @@ if __name__ == '__main__': "changelabels": changelabels }) - runner.run(args.which, args.__dict__, remove="which") \ No newline at end of file + runner.run(args.which, args.__dict__, remove="which")