From 959343795f3b3be3618c68e8e3f96859b9b6a46b Mon Sep 17 00:00:00 2001 From: YAMY <74099316+YAMY1234@users.noreply.github.com> Date: Tue, 6 Jan 2026 01:38:35 -0800 Subject: [PATCH] [test] update acc len threshold to 2.7 for eagle dp attention tests (#16463) --- .../spec/eagle/test_eagle_infer_beta_dp_attention.py | 2 +- .../spec/eagle/test_eagle_infer_beta_dp_attention_large.py | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/test/registered/spec/eagle/test_eagle_infer_beta_dp_attention.py b/test/registered/spec/eagle/test_eagle_infer_beta_dp_attention.py index a0c4d5958..e5494b08f 100644 --- a/test/registered/spec/eagle/test_eagle_infer_beta_dp_attention.py +++ b/test/registered/spec/eagle/test_eagle_infer_beta_dp_attention.py @@ -80,7 +80,7 @@ class TestEagleDPAttnServerSmall(CustomTestCase): def test_a_gsm8k(self): metrics, avg_spec_accept_length = test_gsm8k(self.base_url) self.assertGreater(metrics["accuracy"], 0.64) - self.assertGreater(avg_spec_accept_length, 1.4) + self.assertGreater(avg_spec_accept_length, 2.7) if __name__ == "__main__": diff --git a/test/registered/spec/eagle/test_eagle_infer_beta_dp_attention_large.py b/test/registered/spec/eagle/test_eagle_infer_beta_dp_attention_large.py index 64b055416..fc8c4b12d 100644 --- a/test/registered/spec/eagle/test_eagle_infer_beta_dp_attention_large.py +++ b/test/registered/spec/eagle/test_eagle_infer_beta_dp_attention_large.py @@ -89,8 +89,7 @@ class TestEagleDPAttnServerLarge(CustomTestCase): metrics, avg_spec_accept_length = test_gsm8k(self.base_url) self.assertGreater(metrics["accuracy"], 0.94) - # TODO: Update accept len to 2.04 once the bug is fixed - self.assertGreater(avg_spec_accept_length, 1.4) + self.assertGreater(avg_spec_accept_length, 2.7) if is_in_ci(): write_github_step_summary( f"### test_gsm8k (deepseek-v3-fp4 mtp)\n"