Skip to content

Commit ef7e8d4

Browse files
committed
fix eagle tree test
Signed-off-by: Lu Fang <[email protected]>
1 parent 3676c94 commit ef7e8d4

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/v1/spec_decode/test_eagle.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -664,6 +664,9 @@ def create_deterministic_logits(token_ids, k: int):
664664
# Mock runner for attention metadata building.
665665
proposer.runner = mock.MagicMock()
666666
proposer.runner.attn_groups.append([mock.MagicMock()])
667+
proposer.runner.attn_groups[0][0].metadata_builders = [
668+
attn_metadata_builder
669+
]
667670
proposer.runner.attn_groups[0][0].get_metadata_builder.return_value = \
668671
attn_metadata_builder
669672
proposer._get_attention_metadata_builder = mock.MagicMock(

0 commit comments

Comments
 (0)