mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-07-31 08:00:09 +00:00
fix: error on failed job, do not wait for timeout (#2945)
# What does this PR do? cause post training integration test to error when job fails. ## Test Plan ci
This commit is contained in:
parent
870a37ff4b
commit
c7dc0f21b4
1 changed files with 3 additions and 3 deletions
|
@ -38,9 +38,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
|
||||
#
|
||||
|
||||
|
||||
class TestPostTraining:
|
||||
|
@ -113,6 +112,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