commit

mosaicml · Sep 29, 2023 · a16aea1 · a16aea1
1 parent e896c09
commit a16aea1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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))