Skip to content

Commit

Permalink
remove latex writer since we don't use it (#231)
Browse files Browse the repository at this point in the history
Co-authored-by: Nathan Habib <[email protected]>
  • Loading branch information
clefourrier and NathanHB authored Jul 23, 2024
1 parent 003f05e commit 86fbe64
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 @@ -123,9 +123,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 @@ -144,6 +142,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 86fbe64

Please sign in to comment.