update test

huggingface · Feb 6, 2024 · 3a6922c · 3a6922c
1 parent aefd6d5
commit 3a6922c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,