Skip to content

Commit

Permalink
remove latex writer since we don't use it
Browse files Browse the repository at this point in the history
  • Loading branch information
clefourrier committed Jul 17, 2024
1 parent d43c9a3 commit e5f7f08
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions src/lighteval/evaluator.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@
import copy
from typing import Dict, Union

from pytablewriter import LatexTableWriter, MarkdownTableWriter
from pytablewriter import MarkdownTableWriter

from lighteval.logging.evaluation_tracker import EvaluationTracker
from lighteval.logging.hierarchical_logger import hlog
Expand Down Expand Up @@ -124,9 +124,7 @@ def evaluate( # noqa: C901
def make_results_table(result_dict):
"""Generate table of results."""
md_writer = MarkdownTableWriter()
latex_writer = LatexTableWriter()
md_writer.headers = ["Task", "Version", "Metric", "Value", "", "Stderr"]
latex_writer.headers = ["Task", "Version", "Metric", "Value", "", "Stderr"]

values = []

Expand All @@ -145,6 +143,5 @@ def make_results_table(result_dict):
k = ""
version = ""
md_writer.value_matrix = values
latex_writer.value_matrix = values

return md_writer.dumps()

0 comments on commit e5f7f08

Please sign in to comment.