From 3a6922c4a7041395a2f2b8d5d8991f761f5471ac Mon Sep 17 00:00:00 2001
From: Thomas Wolf <thomas@huggingface.co>
Date: Tue, 6 Feb 2024 18:46:29 +0000
Subject: [PATCH] update test

---
 tests/test_main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_main.py b/tests/test_main.py
index d87d37db4..87016bb77 100644
--- a/tests/test_main.py
+++ b/tests/test_main.py
@@ -2,9 +2,9 @@
 import os
 
 import pytest
-from main import get_parser, main  # noqa: E402
 from pytest import approx
 
+from lighteval.main import get_parser, main  # noqa: E402
 from tests.reference_scores.reference_task_scores import RESULTS_FULL, RESULTS_LITE  # noqa: E402
 from tests.reference_scores.reference_tasks import (  # noqa: E402
     HELM_SUBSET,