From 86fbe6416a94a2ce804f606a07558541c337914c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Cl=C3=A9mentine=20Fourrier?= <22726840+clefourrier@users.noreply.github.com> Date: Tue, 23 Jul 2024 14:33:09 +0200 Subject: [PATCH] remove latex writer since we don't use it (#231) Co-authored-by: Nathan Habib <30601243+NathanHB@users.noreply.github.com> --- src/lighteval/evaluator.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/src/lighteval/evaluator.py b/src/lighteval/evaluator.py index 331b5f386..bd58d11d2 100644 --- a/src/lighteval/evaluator.py +++ b/src/lighteval/evaluator.py @@ -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 @@ -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 = [] @@ -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()