added a comment in the tests & re run tests upstream, works locally

This commit is contained in:
Ubuntu 2025-07-23 18:22:30 +00:00
parent 736404c1bd
commit 41a45580e0

View file

@ -46,6 +46,7 @@ sys.stdout.reconfigure(line_buffering=True)
# -v -s --tb=short --disable-warnings
# SFT test
class TestPostTraining:
@pytest.mark.integration
@pytest.mark.parametrize(
@ -155,6 +156,7 @@ class TestPostTraining:
# assert job_artifacts.checkpoints[0].epoch == 0
# assert "/.llama/checkpoints/Llama3.2-3B-Instruct-sft-0" in job_artifacts.checkpoints[0].path
# DPO test
@pytest.mark.integration
@pytest.mark.parametrize(
"purpose, source",