Gil-Simas commited on
Commit
4e85605
·
1 Parent(s): a90cb74

wandb log to match result structure

Browse files
Files changed (1) hide show
  1. user-friendly-metrics.py +2 -2
user-friendly-metrics.py CHANGED
@@ -176,8 +176,8 @@ class UserFriendlyMetrics(evaluate.Metric):
176
 
177
  # Log overall metrics
178
  if "overall" in result:
179
- for overall_key, overall_metrics in result["overall"].items():
180
- for metric, value in overall_metrics["all"].items():
181
  log_key = (
182
  f"{wandb_section}/overall/{overall_key}/{metric}"
183
  if wandb_section
 
176
 
177
  # Log overall metrics
178
  if "overall" in result:
179
+ for overall_key, overall_metrics in result["overall"]["all"].items():
180
+ for metric, value in overall_metrics.items():
181
  log_key = (
182
  f"{wandb_section}/overall/{overall_key}/{metric}"
183
  if wandb_section