Improve test cases for eagle infer (#7173)

2025-06-13 22:25:13 -07:00
parent 50876abc47
commit ba589b88fc
3 changed files with 44 additions and 32 deletions
--- a/test/srt/test_eagle_infer_a.py
+++ b/test/srt/test_eagle_infer_a.py
@@ -129,7 +129,7 @@ class TestEAGLEEngine(CustomTestCase):
            output["meta_info"]["completion_tokens"]
            / output["meta_info"]["e2e_latency"]
        )
-        print(f"{acc_length=}")
+        print(f"{acc_length=:.4f}, {speed=}")

        if engine.server_args.model_path == DEFAULT_EAGLE_TARGET_MODEL_FOR_TEST:
            self.assertGreater(acc_length, 3.6)