Fix import

Signed-off-by: SumanthRH <[email protected]>

Author: SumanthRH

tests/evals/tasks/test_math.py

@@ -1,5 +1,6 @@
 import pytest
-from skythought_evals.tasks.math.math_handler import MathTaskHandler
+
+from skythought.evals.tasks.math.math_handler import MathTaskHandler
 
 
 class MockTaskConfig:

tests/evals/tasks/test_preprocessing.py

@@ -1,5 +1,6 @@
 import pytest
-from skythought_evals.tasks import MMLUTaskHandler, TaskConfig
+
+from skythought.evals.tasks import MMLUTaskHandler, TaskConfig
 
 SYSTEM_PROMPT = "Please answer the following question:"
 

tests/evals/test_cli.py

@@ -2,12 +2,13 @@
 import os
 
 import pytest
-from skythought_evals.cli import (
+from typer.testing import CliRunner
+
+from skythought.evals.cli import (
     Backend,
     SamplingParameters,
     app,
 )
-from typer.testing import CliRunner
 
 runner = CliRunner(mix_stderr=False)
 
@@ -74,7 +75,7 @@ def test_evaluate_success(tmp_result_dir, mocker):
     """Test a successful execution of the `evaluate` command."""
     # Mock helper functions to avoid actual processing
     mocker.patch(
-        "skythought_evals.cli.parse_common_args",
+        "skythought.evals.cli.parse_common_args",
         return_value=(
             "amc23",
             {},
@@ -90,9 +91,9 @@ def test_evaluate_success(tmp_result_dir, mocker):
             None,
         ),
     )
-    mocker.patch("skythought_evals.cli.get_run_config", return_value={})
-    mocker.patch("skythought_evals.cli.get_output_dir", return_value=tmp_result_dir)
-    mocker.patch("skythought_evals.cli.generate_and_score")
+    mocker.patch("skythought.evals.cli.get_run_config", return_value={})
+    mocker.patch("skythought.evals.cli.get_output_dir", return_value=tmp_result_dir)
+    mocker.patch("skythought.evals.cli.generate_and_score")
 
     # Ensure the result directory does not exist initially
     assert not tmp_result_dir.exists()
@@ -142,7 +143,7 @@ def test_generate_success(tmp_result_dir, mocker):
     """Test a successful execution of the `generate` command."""
     # Mock helper functions to avoid actual processing
     mocker.patch(
-        "skythought_evals.cli.parse_common_args",
+        "skythought.evals.cli.parse_common_args",
         return_value=(
             "amc23",
             {},
@@ -158,9 +159,9 @@ def test_generate_success(tmp_result_dir, mocker):
             None,
         ),
     )
-    mocker.patch("skythought_evals.cli.get_run_config", return_value={})
-    mocker.patch("skythought_evals.cli.get_output_dir", return_value=tmp_result_dir)
-    mocker.patch("skythought_evals.cli.generate_and_save")
+    mocker.patch("skythought.evals.cli.get_run_config", return_value={})
+    mocker.patch("skythought.evals.cli.get_output_dir", return_value=tmp_result_dir)
+    mocker.patch("skythought.evals.cli.generate_and_save")
 
     result = runner.invoke(
         app,

tests/evals/util/test_cli_util.py

@@ -1,5 +1,6 @@
 import pytest
-from skythought_evals.util.cli_util import _parse_multi_args, parse_multi_args
+
+from skythought.evals.util.cli_util import _parse_multi_args, parse_multi_args
 
 
 # Tests for _parse_multi_args

tests/evals/util/test_common.py

@@ -1,5 +1,6 @@
 import pytest
-from skythought_evals.util.common import has_code
+
+from skythought.evals.util.common import has_code
 
 
 @pytest.mark.parametrize(

tests/evals/util/test_math_parsing.py

@@ -1,5 +1,6 @@
 import pytest
-from skythought_evals.util.math_parsing_util import (
+
+from skythought.evals.util.math_parsing_util import (
     choice_answer_clean,
     extract_answer,
     get_multiple_choice_answer,