From b331d79b23d1afc63fc358c3278260996960d6f5 Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Thu, 7 Mar 2024 11:40:29 -0500
Subject: [PATCH] re-enable DPO for tests in modal ci

---
 tests/e2e/test_dpo.py | 3 ---
 1 file changed, 3 deletions(-)

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