phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-26 17:38:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge branch 'main' into feature/dpo-training

Browse source
This commit is contained in:
Ashwin Bharambe 2025-07-30 23:33:00 -07:00 • committed by GitHub
commit 9ac1a01daa
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
419 changed files with 44853 additions and 1661 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tests/integration/post_training/test_post_training.py
View file

@ -25,6 +25,15 @@ logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(
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)
def capture_output(capsys):
"""Fixture to capture and display output during test execution."""
@ -61,6 +70,7 @@ class TestPostTraining:
],
)
@pytest.mark.timeout(360) # 6 minutes timeout
@skip_because_resource_intensive
def test_supervised_fine_tune(self, llama_stack_client, purpose, source):
logger.info("Starting supervised fine-tuning test")