Commit 4e4f0de2 authored by lintangsutawika's avatar lintangsutawika
Browse files

changed term to groups

parent f7d98500
......@@ -267,9 +267,11 @@ def make_table(result_dict, column: str = "results"):
from pytablewriter import MarkdownTableWriter, LatexTableWriter
if column == "results":
column_name = "Task"
elif column == "aggregate":
column_name = "Benchmark"
column_name = "Tasks"
elif column == "groups":
column_name = "Groups"
elif column == "benchmarks":
column_name = "Benchmarks"
md_writer = MarkdownTableWriter()
latex_writer = LatexTableWriter()
......
......@@ -209,8 +209,8 @@ def main() -> None:
f"batch_size: {args.batch_size}{f' ({batch_sizes})' if batch_sizes else ''}"
)
print(evaluator.make_table(results))
if "aggregate" in results:
print(evaluator.make_table(results, "aggregate"))
if "groups" in results:
print(evaluator.make_table(results, "groups"))
if __name__ == "__main__":
......
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