diff --git a/tests/e2e/test_phi.py b/tests/e2e/test_phi.py index 7abed8594..4cc6bcdcc 100644 --- a/tests/e2e/test_phi.py +++ b/tests/e2e/test_phi.py @@ -7,8 +7,6 @@ import os import unittest from pathlib import Path -import pytest - from axolotl.cli import load_datasets from axolotl.common.cli import TrainerCliArgs from axolotl.train import train @@ -21,7 +19,6 @@ LOG = logging.getLogger("axolotl.tests.e2e") os.environ["WANDB_DISABLED"] = "true" -@pytest.mark.skip(reason="doesn't seem to work on modal") class TestPhi(unittest.TestCase): """ Test case for Phi2 models