[Spec] Disambiguate verified_id into bonus_token(s) / accept_tokens (#24724)

2026-05-08 18:24:33 -07:00
parent a61a14f416
commit 1613bae412
18 changed files with 128 additions and 108 deletions
@@ -18,7 +18,7 @@ class TestBuildEagleTree(unittest.TestCase):

    def test_build_tree_kernel_efficient(self):
        """Test the build_tree_kernel_efficient function with known inputs and expected outputs."""
-        verified_id = torch.tensor([29974, 13], device=get_device(), dtype=torch.int32)
+        bonus_tokens = torch.tensor([29974, 13], device=get_device(), dtype=torch.int32)
        score_list = [
            torch.tensor(
                [
@@ -244,7 +244,7 @@ class TestBuildEagleTree(unittest.TestCase):
            retrieve_next_sibling,
            draft_tokens,
        ) = build_tree_kernel_efficient(
-            verified_id=verified_id,
+            bonus_tokens=bonus_tokens,
            parent_list=parent_list,
            top_scores_index=top_scores_index,
            draft_tokens=draft_tokens,