mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-08-01 16:24:44 +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:
|
# How to run this test:
|
||||||
#
|
#
|
||||||
# pytest llama_stack/providers/tests/post_training/test_post_training.py
|
# LLAMA_STACK_CONFIG=ci-tests uv run --dev pytest tests/integration/post_training/test_post_training.py
|
||||||
# -m "torchtune_post_training_huggingface_datasetio"
|
#
|
||||||
# -v -s --tb=short --disable-warnings
|
|
||||||
|
|
||||||
|
|
||||||
class TestPostTraining:
|
class TestPostTraining:
|
||||||
|
@ -113,6 +112,7 @@ class TestPostTraining:
|
||||||
break
|
break
|
||||||
|
|
||||||
logger.info(f"Current status: {status}")
|
logger.info(f"Current status: {status}")
|
||||||
|
assert status.status in ["scheduled", "in_progress", "completed"]
|
||||||
if status.status == "completed":
|
if status.status == "completed":
|
||||||
break
|
break
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue