re-enable DPO for tests in modal ci

axolotl-ai-cloud · Mar 7, 2024 · b331d79 · b331d79
1 parent 638c2da
commit b331d79
Showing 1 changed file with 0 additions and 3 deletions.
diff --git 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