From a16aea1bc5bc6925116bf1fb2edcd6363c477f97 Mon Sep 17 00:00:00 2001 From: Jeremy Dohmann Date: Fri, 29 Sep 2023 16:03:44 -0400 Subject: [PATCH] commit --- scripts/eval/eval.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/eval/eval.py b/scripts/eval/eval.py index 09db0a870a..7e2509cb30 100644 --- a/scripts/eval/eval.py +++ b/scripts/eval/eval.py @@ -322,7 +322,7 @@ def main(cfg: DictConfig): print( eval_gauntlet_df.sort_values( - next(eval_gauntlet_callback.averages.keys()), ascending=False).to_markdown(index=False)) + list(eval_gauntlet_callback.averages.keys())[0], ascending=False).to_markdown(index=False)) print(f'Printing complete results for all models') assert models_df is not None print(models_df.to_markdown(index=False))