chore(ci): disable post training tests

This commit is contained in:
Ashwin Bharambe 2025-07-29 13:58:35 -07:00
parent 2e5ca3f15c
commit 95d715b401

View file

@ -22,6 +22,15 @@ logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
skip_because_resource_intensive = pytest.mark.skip(
reason="""
Post training tests are extremely resource intensive. They download large models and partly as a result,
are very slow to run. We cannot run them on every single PR update. CI should be considered
a scarce resource and properly utilitized.
"""
)
@pytest.fixture(autouse=True) @pytest.fixture(autouse=True)
def capture_output(capsys): def capture_output(capsys):
"""Fixture to capture and display output during test execution.""" """Fixture to capture and display output during test execution."""
@ -57,6 +66,7 @@ class TestPostTraining:
], ],
) )
@pytest.mark.timeout(360) # 6 minutes timeout @pytest.mark.timeout(360) # 6 minutes timeout
@skip_because_resource_intensive
def test_supervised_fine_tune(self, llama_stack_client, purpose, source): def test_supervised_fine_tune(self, llama_stack_client, purpose, source):
logger.info("Starting supervised fine-tuning test") logger.info("Starting supervised fine-tuning test")