Skip to content

GitLab

Commit 61d0f137 authored by lintangsutawika's avatar lintangsutawika
Browse files

use all_headers for headers

parent 19738ed2
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 13 deletions
lm_eval/utils.py
View file @ 61d0f137
...@@ -280,19 +280,7 @@ def make_table(result_dict, column: str = "results"): ...@@ -280,19 +280,7 @@ def make_table(result_dict, column: str = "results"):
elif column == "groups": elif column == "groups":
column_name = "Groups" column_name = "Groups"
md_writer = MarkdownTableWriter() all_headers = [
latex_writer = LatexTableWriter()
md_writer.headers = [
column_name,
"Version",
"Filter",
"n-shot",
"Metric",
"Value",
"",
"Stderr",
]
latex_writer.headers = [
column_name, column_name,
"Version", "Version",
"Filter", "Filter",
...@@ -303,6 +291,11 @@ def make_table(result_dict, column: str = "results"): ...@@ -303,6 +291,11 @@ def make_table(result_dict, column: str = "results"):
"Stderr", "Stderr",
] ]
md_writer = MarkdownTableWriter()
latex_writer = LatexTableWriter()
md_writer.headers = all_headers
latex_writer.headers = all_headers
values = [] values = []
for k, dic in result_dict[column].items(): for k, dic in result_dict[column].items():
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment