mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-26 08:31:59 +00:00
Merge branch 'main' into feature/dpo-training
This commit is contained in:
commit
b68b818539
265 changed files with 10254 additions and 7796 deletions
|
|
@ -41,9 +41,8 @@ sys.stdout.reconfigure(line_buffering=True)
|
|||
|
||||
# How to run this test:
|
||||
#
|
||||
# pytest llama_stack/providers/tests/post_training/test_post_training.py
|
||||
# -m "torchtune_post_training_huggingface_datasetio"
|
||||
# -v -s --tb=short --disable-warnings
|
||||
# LLAMA_STACK_CONFIG=ci-tests uv run --dev pytest tests/integration/post_training/test_post_training.py
|
||||
#
|
||||
|
||||
|
||||
# SFT test
|
||||
|
|
@ -117,6 +116,7 @@ class TestPostTraining:
|
|||
break
|
||||
|
||||
logger.info(f"Current status: {status}")
|
||||
assert status.status in ["scheduled", "in_progress", "completed"]
|
||||
if status.status == "completed":
|
||||
break
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue