re-enable phi for tests in modal ci (#1373)
Browse files- tests/e2e/test_phi.py +0 -3
tests/e2e/test_phi.py
CHANGED
@@ -7,8 +7,6 @@ import os
|
|
7 |
import unittest
|
8 |
from pathlib import Path
|
9 |
|
10 |
-
import pytest
|
11 |
-
|
12 |
from axolotl.cli import load_datasets
|
13 |
from axolotl.common.cli import TrainerCliArgs
|
14 |
from axolotl.train import train
|
@@ -21,7 +19,6 @@ LOG = logging.getLogger("axolotl.tests.e2e")
|
|
21 |
os.environ["WANDB_DISABLED"] = "true"
|
22 |
|
23 |
|
24 |
-
@pytest.mark.skip(reason="doesn't seem to work on modal")
|
25 |
class TestPhi(unittest.TestCase):
|
26 |
"""
|
27 |
Test case for Phi2 models
|
|
|
7 |
import unittest
|
8 |
from pathlib import Path
|
9 |
|
|
|
|
|
10 |
from axolotl.cli import load_datasets
|
11 |
from axolotl.common.cli import TrainerCliArgs
|
12 |
from axolotl.train import train
|
|
|
19 |
os.environ["WANDB_DISABLED"] = "true"
|
20 |
|
21 |
|
|
|
22 |
class TestPhi(unittest.TestCase):
|
23 |
"""
|
24 |
Test case for Phi2 models
|