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"]
|
batches = [l for l in lines if l["type"] == "batch"]
|
||||||
complete = lines[-1]
|
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
|
assert complete["total"] == 200
|
||||||
|
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue