Skip to content

Commit 0ebc186

Browse files
committed
bug fix on comparison
1 parent e9022c4 commit 0ebc186

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

compare_benchmarks.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -61,14 +61,13 @@ def group_results_by_config(results: List[Dict[str, Any]]) -> Dict[str, Dict[str
6161
config_key = f"{dataset}|"
6262
config_key += "|".join([f"{k}={v}" for k, v in sorted(params.items())])
6363

64-
# Extract the metric name and value
65-
metric_name, metric_value = next(iter(metrics.items()))
66-
67-
# Store the metric value under this configuration
64+
# Store all metrics for this configuration
6865
if config_key not in grouped_results:
6966
grouped_results[config_key] = {}
7067

71-
grouped_results[config_key][metric_name] = float(metric_value)
68+
# Process all metrics instead of just the first one
69+
for metric_name, metric_value in metrics.items():
70+
grouped_results[config_key][metric_name] = float(metric_value)
7271

7372
return grouped_results
7473

0 commit comments

Comments
 (0)