Skip to content

Commit 1d1bf86

Browse files
committed
Fix filewise metric encoding
Signed-off-by: Ryan <rlangman@nvidia.com>
1 parent 64c918e commit 1d1bf86

2 files changed

Lines changed: 5 additions & 3 deletions

File tree

examples/tts/magpietts_inference.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -338,11 +338,13 @@ def run_inference_and_evaluation(
338338
filewise_metrics_all_repeats.extend(filewise_metrics)
339339

340340
# Save metrics
341-
with open(os.path.join(eval_dir, f"{dataset}_metrics_{repeat_idx}.json"), "w") as f:
341+
metrics_path = os.path.join(eval_dir, f"{dataset}_metrics_{repeat_idx}.json")
342+
with open(metrics_path, "w") as f:
342343
json.dump(metrics, f, indent=4)
343344

344345
sorted_filewise = sorted(filewise_metrics, key=lambda x: x.get('cer', 0), reverse=True)
345-
with open(os.path.join(eval_dir, f"{dataset}_filewise_metrics_{repeat_idx}.json"), "w") as f:
346+
filewise_metrics_path = os.path.join(eval_dir, f"{dataset}_filewise_metrics_{repeat_idx}.json")
347+
with open(filewise_metrics_path, "w", encoding="utf-8") as f:
346348
json.dump(sorted_filewise, f, indent=4, ensure_ascii=False)
347349

348350
# Append to per-run CSV

scripts/tts_comparison_report/reporting/helpers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414
from datetime import UTC, datetime, timedelta
1515
from pathlib import Path
1616

17-
from scripts.tts_comparison_report.reporting.constants import DUMMY_TASK_ID, TICKET_URL_PREFIX
17+
from scripts.tts_comparison_report.reporting.constants import TICKET_URL_PREFIX
1818
from scripts.tts_comparison_report.reporting.models import ExpirationInfo, TaskInfo
1919

2020

0 commit comments

Comments
 (0)