diff --git a/tests/e2e/test_dpo.py b/tests/e2e/test_dpo.py index e28df7411f..ac3c6d0693 100644 --- a/tests/e2e/test_dpo.py +++ b/tests/e2e/test_dpo.py @@ -7,8 +7,6 @@ import unittest from pathlib import Path -import pytest - from axolotl.cli import load_rl_datasets from axolotl.common.cli import TrainerCliArgs from axolotl.train import train @@ -21,7 +19,6 @@ os.environ["WANDB_DISABLED"] = "true" -@pytest.mark.skip(reason="doesn't seem to work on modal") class TestDPOLlamaLora(unittest.TestCase): """ Test case for DPO Llama models using LoRA