Fix e2e test for adaptive first batch size
test_large_batch_streaming expected 4 batches (200/50) but the adaptive first batch splits the first batch into 5+45, producing 5 batches total.
This commit is contained in:
parent
8ef6868881
commit
c6f7b47446
1 changed files with 1 additions and 1 deletions
|
|
@ -80,7 +80,7 @@ async def test_large_batch_streaming(
|
|||
batches = [l for l in lines if l["type"] == "batch"]
|
||||
complete = lines[-1]
|
||||
|
||||
assert len(batches) == 4
|
||||
assert len(batches) == 5 # first batch is smaller (FIRST_BATCH_SIZE=5), then 4x50
|
||||
assert complete["total"] == 200
|
||||
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue