From 41a45580e0ac54213f30ea3362f396872edf6771 Mon Sep 17 00:00:00 2001
From: Ubuntu <ubuntu@ip-172-31-43-83.ec2.internal>
Date: Wed, 23 Jul 2025 18:22:30 +0000
Subject: [PATCH] added a comment in the tests & re run tests upstream, works
 locally

---
 tests/integration/post_training/test_post_training.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/integration/post_training/test_post_training.py b/tests/integration/post_training/test_post_training.py
index 2ead02a70..4c1b3e005 100644
--- a/tests/integration/post_training/test_post_training.py
+++ b/tests/integration/post_training/test_post_training.py
@@ -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",