Split the __init__ of scheduler as smaller functions. Improve the eagle tests (#4128)

2025-03-06 00:13:20 -08:00
parent 0804dd11a0
commit fcc2e37f69
7 changed files with 279 additions and 341 deletions
--- a/test/srt/test_moe_ep.py
+++ b/test/srt/test_moe_ep.py
@@ -94,7 +94,7 @@ class TestEpMoEFP8(unittest.TestCase):
        )

        metrics = run_eval(args)
-        assert metrics["score"] >= 0.5
+        self.assertGreaterEqual(metrics["score"], 0.5)

    def test_mgsm_en(self):
        args = SimpleNamespace(
@@ -106,7 +106,7 @@ class TestEpMoEFP8(unittest.TestCase):
        )

        metrics = run_eval(args)
-        assert metrics["score"] >= 0.8
+        self.assertGreaterEqual(metrics["score"], 0.8)


 if __name__ == "__main__":