[Misc] Make PlaceholderRange.get_num_embeds a method (#34035)

Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
This commit is contained in:
Cyrus Leung
2026-02-07 13:30:17 +08:00
committed by GitHub
parent bc32444b23
commit 48312e579a
9 changed files with 11 additions and 12 deletions

View File

@@ -19,7 +19,7 @@ from vllm.multimodal.inputs import PlaceholderRange
def test_placeholder_range_get_num_embeds(is_embed, expected):
length = len(is_embed) if is_embed is not None else 5
pr = PlaceholderRange(offset=0, length=length, is_embed=is_embed)
assert pr.get_num_embeds == expected
assert pr.get_num_embeds() == expected
@pytest.mark.parametrize(