From e349b2170231b663e78a8059cb668d6d14d30f90 Mon Sep 17 00:00:00 2001
From: EmmonsCurse <1577972691@qq.com>
Date: Sun, 19 Apr 2026 18:53:17 +0800
Subject: [PATCH] [CI] Align logprobs test baselines with Paddle Update

---
 .github/workflows/_logprob_test_linux.yml            | 2 +-
 tests/e2e/4cards_cases/test_ernie_21b_tp1_dp4.py     | 8 ++++----
 tests/e2e/4cards_cases/test_ernie_21b_tp1_dp4_mtp.py | 8 ++++----
 tests/e2e/utils/baseline_manager.py                  | 2 +-
 4 files changed, 10 insertions(+), 10 deletions(-)

diff --git a/.github/workflows/_logprob_test_linux.yml b/.github/workflows/_logprob_test_linux.yml
index 0a014d26854..f3e8b9e66e5 100644
--- a/.github/workflows/_logprob_test_linux.yml
+++ b/.github/workflows/_logprob_test_linux.yml
@@ -218,7 +218,7 @@ jobs:
             -d "{\"messages\": [{\"role\": \"user\", \"content\": \"1+1=?\"}], \"logprobs\": true}"
           set +e
           rm -rf ./baseline_output
-          cp -r baseline_dev_0311/ERNIE-4.5-0.3B-Paddle ./baseline_output
+          cp -r baseline_0419/ERNIE-4.5-0.3B-Paddle ./baseline_output
           LOGPROB_EXIT_CODE=0
           python3.10 lanucher.py --request_template TOKEN_LOGPROB --url http://localhost:${FD_API_PORT}/v1/chat/completions  --case ./cases/demo.yaml  --concurrency 1 --name demo --exe logprob || LOGPROB_EXIT_CODE=$?
           echo "LOGPROB_EXIT_CODE=${LOGPROB_EXIT_CODE}" > /workspace/exit_code.env
diff --git a/tests/e2e/4cards_cases/test_ernie_21b_tp1_dp4.py b/tests/e2e/4cards_cases/test_ernie_21b_tp1_dp4.py
index 4fb178d4582..6da27a1b976 100644
--- a/tests/e2e/4cards_cases/test_ernie_21b_tp1_dp4.py
+++ b/tests/e2e/4cards_cases/test_ernie_21b_tp1_dp4.py
@@ -609,9 +609,9 @@ def test_non_stream_with_logprobs(api_url):
 
     base_path = os.getenv("MODEL_PATH")
     if base_path:
-        base_file = os.path.join(base_path, "21b_tp1_dp4_logprobs_non_stream_static_baseline.txt")
+        base_file = os.path.join(base_path, "21b_tp1_dp4_logprobs_non_stream_static_baseline_0419.txt")
     else:
-        base_file = "21b_tp1_dp4_logprobs_non_stream_static_baseline.txt"
+        base_file = "21b_tp1_dp4_logprobs_non_stream_static_baseline_0419.txt"
 
     with open(base_file, "r", encoding="utf-8") as f:
         baseline = json.load(f)
@@ -647,9 +647,9 @@ def test_stream_with_logprobs(api_url):
 
     base_path = os.getenv("MODEL_PATH")
     if base_path:
-        base_file = os.path.join(base_path, "21b_tp1_dp4_logprobs_stream_static_baseline.txt")
+        base_file = os.path.join(base_path, "21b_tp1_dp4_logprobs_stream_static_baseline_0419.txt")
     else:
-        base_file = "21b_tp1_dp4_logprobs_stream_static_baseline.txt"
+        base_file = "21b_tp1_dp4_logprobs_stream_static_baseline_0419.txt"
 
     with open(base_file, "r", encoding="utf-8") as f:
         baseline = json.load(f)
diff --git a/tests/e2e/4cards_cases/test_ernie_21b_tp1_dp4_mtp.py b/tests/e2e/4cards_cases/test_ernie_21b_tp1_dp4_mtp.py
index 6e4e36f7392..294ed055215 100644
--- a/tests/e2e/4cards_cases/test_ernie_21b_tp1_dp4_mtp.py
+++ b/tests/e2e/4cards_cases/test_ernie_21b_tp1_dp4_mtp.py
@@ -516,9 +516,9 @@ def test_non_stream_with_logprobs(api_url):
     base_path = os.getenv("MODEL_PATH")
 
     if base_path:
-        base_file = os.path.join(base_path, "21b_tp1_dp4_mtp_logprobs_non_stream_static_baseline.txt")
+        base_file = os.path.join(base_path, "21b_tp1_dp4_mtp_logprobs_non_stream_static_baseline_0419.txt")
     else:
-        base_file = "21b_tp1_dp4_mtp_logprobs_non_stream_static_baseline.txt"
+        base_file = "21b_tp1_dp4_mtp_logprobs_non_stream_static_baseline_0419.txt"
 
     with open(base_file, "r", encoding="utf-8") as f:
         baseline = json.load(f)
@@ -555,9 +555,9 @@ def test_stream_with_logprobs(api_url):
     base_path = os.getenv("MODEL_PATH")
 
     if base_path:
-        base_file = os.path.join(base_path, "21b_tp1_dp4_mtp_logprobs_stream_static_baseline.txt")
+        base_file = os.path.join(base_path, "21b_tp1_dp4_mtp_logprobs_stream_static_baseline_0419.txt")
     else:
-        base_file = "21b_tp1_dp4_mtp_logprobs_stream_static_baseline.txt"
+        base_file = "21b_tp1_dp4_mtp_logprobs_stream_static_baseline_0419.txt"
 
     with open(base_file, "r", encoding="utf-8") as f:
         baseline = json.load(f)
diff --git a/tests/e2e/utils/baseline_manager.py b/tests/e2e/utils/baseline_manager.py
index ce3338eb9f8..d3c9fb43e46 100644
--- a/tests/e2e/utils/baseline_manager.py
+++ b/tests/e2e/utils/baseline_manager.py
@@ -12,7 +12,7 @@ def __init__(self, base_dir="/ModelData"):
         self.base_dir = base_dir
 
     def _get_path(self, name: str):
-        branch = os.getenv("TEST_BRANCH", "default")
+        branch = os.getenv("TEST_BRANCH", "0419")
         if os.getenv("STATIC_C8") == "1":
             c8_mode = "_static_c8"
         elif os.getenv("DYNAMIC_C8") == "1":