From c6800a36d755943ce110e9d0b3e5578ee6c6597a Mon Sep 17 00:00:00 2001 From: T5 Team Date: Mon, 14 Aug 2023 13:56:01 -0700 Subject: [PATCH] No public description PiperOrigin-RevId: 556902744 --- 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 bd21be67..9cd0340d 100644 --- a/t5/evaluation/eval_utils.py +++ b/t5/evaluation/eval_utils.py @@ -238,7 +238,7 @@ def metric_group_max(df, metric_names=None): group_max_step = group_df.fillna(-1e9).idxmax(axis=0) metric_max = pd.Series() metric_max_step = pd.Series() - for group_name, max_step in group_max_step.iteritems(): + for group_name, max_step in group_max_step.items(): for metric in group_to_metrics[group_name]: metric_max[metric] = df[metric][max_step] metric_max_step[metric] = max_step