Fix manual chunked-prefill test to use req.fill_len after fill_ids refactor (#29032)
This commit is contained in:
@@ -476,12 +476,10 @@ class TestSpecialCaseBasic(ScriptedTestCase):
|
|||||||
and req.extend_input_len is not None
|
and req.extend_input_len is not None
|
||||||
):
|
):
|
||||||
saw_mid_chunk = True
|
saw_mid_chunk = True
|
||||||
assert (
|
assert req.fill_len == len(req.prefix_indices) + req.extend_input_len, (
|
||||||
len(req.fill_ids) == len(req.prefix_indices) + req.extend_input_len
|
|
||||||
), (
|
|
||||||
f"init_next_round_input must rebuild fill_ids to the committed "
|
f"init_next_round_input must rebuild fill_ids to the committed "
|
||||||
f"prefix plus the in-flight chunk; "
|
f"prefix plus the in-flight chunk; "
|
||||||
f"fill_ids_len={len(req.fill_ids)}, "
|
f"fill_ids_len={req.fill_len}, "
|
||||||
f"prefix_indices_len={len(req.prefix_indices)}, "
|
f"prefix_indices_len={len(req.prefix_indices)}, "
|
||||||
f"extend_input_len={req.extend_input_len}, "
|
f"extend_input_len={req.extend_input_len}, "
|
||||||
f"chunks_done={r.chunks_done}"
|
f"chunks_done={r.chunks_done}"
|
||||||
|
|||||||
Reference in New Issue
Block a user