use all_headers for headers

61d0f137 · lintangsutawika · 19738ed2 · 61d0f137
Commit 61d0f137 authored Nov 28, 2023 by lintangsutawika
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 13 deletions

lm_eval/utils.py lm_eval/utils.py +6 -13

No files found.
--- a/lm_eval/utils.py
+++ b/lm_eval/utils.py
@@ -280,19 +280,7 @@ def make_table(result_dict, column: str = "results"):
    elif column == "groups":
        column_name = "Groups"

-    md_writer = MarkdownTableWriter()
-    latex_writer = LatexTableWriter()
-    md_writer.headers = [
-        column_name,
-        "Version",
-        "Filter",
-        "n-shot",
-        "Metric",
-        "Value",
-        "",
-        "Stderr",
-    ]
-    latex_writer.headers = [
+    all_headers = [
        column_name,
        "Version",
        "Filter",
@@ -303,6 +291,11 @@ def make_table(result_dict, column: str = "results"):
        "Stderr",
    ]

+    md_writer = MarkdownTableWriter()
+    latex_writer = LatexTableWriter()
+    md_writer.headers = all_headers
+    latex_writer.headers = all_headers
+
    values = []

    for k, dic in result_dict[column].items():