[Spec] Extract the shared draft() tail into build_eagle_verify_input (#31375)

2026-07-15 15:59:31 -07:00
parent 7a973c03a0
commit b0b2dfbda1
4 changed files with 114 additions and 124 deletions
@@ -192,7 +192,7 @@ class TestEagleWorkerV2BackendFallback(CustomTestCase):
                )

                with patch(
-                    "sglang.srt.speculative.eagle_worker_v2.build_tree_kernel_efficient",
+                    "sglang.srt.speculative.eagle_worker_common.build_tree_kernel_efficient",
                    return_value=tree_result,
                ), patch(
                    "sglang.srt.speculative.eagle_worker_v2.prepare_for_draft",