From 93fb559c16987d1f3fe27820c6f3bdc12f30e149 Mon Sep 17 00:00:00 2001
From: T5 Team <no-reply@google.com>
Date: Tue, 26 Sep 2023 12:43:23 -0700
Subject: [PATCH] No public description

PiperOrigin-RevId: 568620982
---
 t5/evaluation/eval_utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/t5/evaluation/eval_utils.py b/t5/evaluation/eval_utils.py
index 9cd0340d..791eb024 100644
--- a/t5/evaluation/eval_utils.py
+++ b/t5/evaluation/eval_utils.py
@@ -233,7 +233,7 @@ def metric_group_max(df, metric_names=None):
   for group, metrics in group_to_metrics.items():
     if not all(m in df for m in metrics):
       continue
-    group_df[group] = df[metrics].mean(axis=1)
+    group_df[group] = df[list(metrics)].mean(axis=1)
   # Need to replace nan with large negative value for idxmax
   group_max_step = group_df.fillna(-1e9).idxmax(axis=0)
   metric_max = pd.Series()