From 1a57145975570c51fb1d735ad0a236682cdc7ac9 Mon Sep 17 00:00:00 2001 From: shuwenn <47200617+alphabetc1@users.noreply.github.com> Date: Thu, 4 Jun 2026 10:07:17 +0800 Subject: [PATCH] [codex] Fix adaptive metrics test flake (#27135) --- test/registered/spec/eagle/test_adaptive_speculative.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/test/registered/spec/eagle/test_adaptive_speculative.py b/test/registered/spec/eagle/test_adaptive_speculative.py index 6757b1e11..e301daa1c 100644 --- a/test/registered/spec/eagle/test_adaptive_speculative.py +++ b/test/registered/spec/eagle/test_adaptive_speculative.py @@ -194,9 +194,11 @@ class TestAdaptiveSpeculativeServer(CustomTestCase): steps = self._scrape_metric("sglang:spec_num_steps") draft_tokens = self._scrape_metric("sglang:spec_num_draft_tokens") - self.assertEqual(steps, 3.0, "spec_num_steps gauge missing or wrong") - self.assertEqual( - draft_tokens, 4.0, "spec_num_draft_tokens gauge missing or wrong" + self.assertIn(steps, {1.0, 3.0}, "spec_num_steps gauge has unexpected value") + self.assertIn( + draft_tokens, + {2.0, 4.0}, + "spec_num_draft_tokens gauge has unexpected value", )