Skip to content

Commit cf9aaa8

Browse files
authored
Adapt logprobs baseline update in test_ernie_21b_mtp_multistep.py
1 parent ccca007 commit cf9aaa8

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

tests/e2e/test_ernie_21b_mtp_multistep.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -212,11 +212,11 @@ def test_prefix_cache_text(api_url):
212212
if os.getenv("BASELINE") == "1":
213213
baseline_manager.save("base_21b_step3", result)
214214
baseline_manager.save("base_21b_mtp_metrics_step3", speculate_metrics_2)
215-
baseline_manager.save("base_21b_logprobs_step3", logprobs_2)
215+
baseline_manager.save("base_21b_logprobs_step3_new", logprobs_2)
216216

217217
baseline_result = baseline_manager.load("base_21b_step3")
218218
baseline_mtp_metrics = baseline_manager.load("base_21b_mtp_metrics_step3")
219-
baseline_logprobs = baseline_manager.load("base_21b_logprobs_step3")
219+
baseline_logprobs = baseline_manager.load("base_21b_logprobs_step3_new")
220220

221221
assert logprobs == logprobs_2, (
222222
"logprobs 前后不一致\n"

0 commit comments

Comments
 (0)