From 598c96504388f7d92e70965590b03a06c6edf16c Mon Sep 17 00:00:00 2001
From: Wing Lian <wing@axolotl.ai>
Date: Sun, 22 Mar 2026 12:00:55 -0400
Subject: [PATCH] use train_loss for sp test

---
 tests/e2e/multigpu/patched/test_sp.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/e2e/multigpu/patched/test_sp.py b/tests/e2e/multigpu/patched/test_sp.py
index ba0636879..a005e6742 100644
--- a/tests/e2e/multigpu/patched/test_sp.py
+++ b/tests/e2e/multigpu/patched/test_sp.py
@@ -94,7 +94,7 @@ class TestSequenceParallelism:
 
         check_tensorboard(
             temp_dir + "/runs",
-            "train/loss",
+            "train/train_loss",
             threshold,
             "Train Loss (%s) is too high",
         )